Updated Streamlit frontend

backend/__init__.py

@@ -7,15 +7,18 @@ from fastapi.exceptions import HTTPException
 from backend.utils import DBConnection
 from backend.core.ConfigEnv import config
 
-from langchain.llms import CTransformers, Clarifai
+from langchain.llms import Clarifai
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate 
 
+import sys
+
 app = FastAPI(title="Techdocs",
               version="V0.0.1",
               description="API for automatic code documentation generation!"
               )
 
+print(sys.getsizeof(app))
 from backend import router
 
 try:
@@ -36,13 +39,19 @@ try:
         model_version_id=config.MODEL_VERSION_ID,
     )
 
+    print(sys.getsizeof(llm))
+
     llmchain = LLMChain(
         prompt=prompt,
         llm=llm
     )
 
+    print(sys.getsizeof(llmchain))
+
     app.state.llmchain = llmchain
 
+    print(sys.getsizeof(app))
+
 except mysql.connector.Error as err:
     raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(err))    
 

backend/services/auth/ops.py

@@ -1,6 +1,5 @@
 from .utils.auth_funcs import *
 from .utils.JWTBearer import *
-from backend.services import parser
 from backend.models import *
 from backend.services.db.utils.DBQueries import DBQueries
 from backend.core.Exceptions import *

backend/utils/__init__.py

@@ -0,0 +1 @@
+from .DBConnection import DBConnection

frontend/Login.py

@@ -11,29 +11,9 @@ def auth_page():
 
     headers={"accept":"application/json"}
 
-    tab1, tab2 = st.tabs(["Signup", "Login"])
+    tab1, tab2 = st.tabs(["Login", "Signup"])
 
     with tab1:
-        with st.form(key="myform1"):
-            username = st.text_input(label="Username", label_visibility="collapsed", placeholder="Username")
-            password = st.text_input(label="Password", label_visibility="collapsed", placeholder="Password", type="password")
-            email = st.text_input(label="Email", label_visibility="collapsed", placeholder="Email")
-            signup_button = st.form_submit_button(label="Signup")
-
-        with st.spinner("Signing up..."):
-            if signup_button:
-                try:
-                    credentials = {"username":username, "password":password, "email":email}
-                    response = requests.post(url=base_url + "/auth/signup", headers=headers, data=json.dumps(credentials))
-                    if (response.status_code!=200):
-                        raise Exception("Signup Failed")
-                
-                    st.success("Signed up successfully")
-                except:
-                    st.error("Signup Failed")       
-            
-
-    with tab2:
         with st.form(key="myform2"):
             username = st.text_input(label="Username", label_visibility="collapsed", placeholder="Username")
             password = st.text_input(label="Password", label_visibility="collapsed", placeholder="Password", type="password")
@@ -56,4 +36,25 @@ def auth_page():
                 except Exception as e:
                     st.error(e)
 
+    with tab2:
+        with st.form(key="myform1"):
+            username = st.text_input(label="Username", label_visibility="collapsed", placeholder="Username")
+            password = st.text_input(label="Password", label_visibility="collapsed", placeholder="Password", type="password")
+            email = st.text_input(label="Email", label_visibility="collapsed", placeholder="Email")
+            signup_button = st.form_submit_button(label="Signup")
+
+        with st.spinner("Signing up..."):
+            if signup_button:
+                try:
+                    credentials = {"username":username, "password":password, "email":email}
+                    response = requests.post(url=base_url + "/auth/signup", headers=headers, data=json.dumps(credentials))
+                    if (response.status_code!=200):
+                        raise Exception("Signup Failed")
+                
+                    st.success("Signed up successfully")
+                except:
+                    st.error("Signup Failed")       
+            
+
+
 

frontend/pages/Code.py

@@ -69,9 +69,8 @@ def code_page():
     if st.button("Generate Comment"):
         if code_input:
             headers['Authorization'] = f"Bearer {st.session_state.access_token}"
-            response = query_post(base_url + '/api/inference', headers=headers, params={'code_block':"def add(a,b):\n\treturn a+b", 'api_key':API_KEY})
+            response = query_post(base_url + '/api/inference', headers=headers, params={'code_block':code_input, 'api_key':API_KEY})
             docstr = response.json()["docstr"]
-            print(docstr)
             comment_placeholder.subheader("Generated Comment:")
             comment_placeholder.markdown(f"<pre><code>{docstr}</code></pre>", unsafe_allow_html=True)
             # Scroll to the comment section

frontend/🏡 Home.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from Login import auth_page
 from PIL import Image 
+import textwrap
 
 import base64
 
@@ -18,11 +19,7 @@ def get_base64_bin_file(bin_file):
         data = f.read()
     return base64.b64encode(data).decode()
 
-
-# image = Image.open('assets/poster.jpg')
-# st.image(image, caption='ELIGILOAN')
-
-st.markdown("# :DocGup-tea: AI based Documentation Generator 📃")
+st.markdown("# DocGup-tea: AI based Documentation Generator 📃")
 
 def logout():
     del st.session_state["access_token"]