Merge pull request #5 from HemanthSai7/main

backend/services/db/utils/DBQueries.py

@@ -1,8 +1,3 @@
-# fix ObjectId & FastApi conflict
-import pydantic
-from bson.objectid import ObjectId
-pydantic.json.ENCODERS_BY_TYPE[ObjectId]=str
-
 from typing import Union, Tuple, List
 
 from backend.utils import DBConnection

frontend/Login.py

@@ -6,7 +6,7 @@ import streamlit as st
 
 def auth_page():
 
-    base_url = 'http://127.0.0.1:8000'
+    base_url = 'https://hemanthsai7-techdocsapi.hf.space'
 
 
     headers={"accept":"application/json"}

frontend/pages/Code.py

@@ -4,8 +4,6 @@ from PIL import Image
 
 import streamlit as st
 from Login import auth_page
-# frontend.Login import auth_pageom from streamlit_extras import 
-
 
 st.set_page_config(
     page_title="Welcome",
@@ -14,7 +12,7 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-st.write("# DocGup-tea: AI based Documentation Generator 💀")
+st.write("# Welcome to Techdocs: Where Code Meets Clarity! 🚀")
 
 def logout():
     del st.session_state["access_token"]
@@ -30,7 +28,7 @@ with st.sidebar:
         if st.button("Generate API KEY"):
             with st.spinner("Generating API Key..."):
                 try:
-                    base_url = "http://localhost:8000"
+                    base_url = "https://hemanthsai7-techdocsapi.hf.space"
                     headers={"accept":"application/json", "Authorization": f"Bearer {st.session_state.access_token}"}
                     response = requests.put(url=base_url + "/auth/regenerate_api_key", headers=headers, data=json.dumps({"username":st.session_state.username}))
                     if (response.status_code!=200):
@@ -50,7 +48,7 @@ with st.sidebar:
 
 
 def code_page():
-    base_url = 'http://localhost:8000'
+    base_url = 'https://hemanthsai7-techdocsapi.hf.space'
 
     def query_post(url, headers, data=None, params=None):
         response = requests.post(url, data=data, headers=headers, params=params)
@@ -58,15 +56,14 @@ def code_page():
     
     headers={"accept":"application/json"}
 
-
+    st.subheader("Enter your API key to generate documentation.")
     API_KEY = st.text_input(label="Enter your API key", label_visibility="hidden",placeholder="Enter your API key", type="password")
-    st.title("Code Documentation Generator")
-    st.write("Enter your code and click 'Generate Comment' to get the corresponding comment.")
+    st.subheader("Enter your code and click 'Generate Documentation' to get the corresponding comment.")
 
-    code_input = st.text_area("Code Input", height=200)
+    code_input = st.text_area("Code Input", height=300)
     comment_placeholder = st.empty()
 
-    if st.button("Generate Comment"):
+    if st.button("Generate Documentation"):
         if code_input:
             headers['Authorization'] = f"Bearer {st.session_state.access_token}"
             response = query_post(base_url + '/api/inference', headers=headers, params={'code_block':code_input, 'api_key':API_KEY})

requirements.txt

@@ -1,10 +1,11 @@
 fastapi==0.99.1
 uvicorn
 requests
+langchain
+clarifai
+Pillow
 pydantic==1.10.12
 python-jose[cryptography]
 passlib[bcrypt]
 mysql-connector-python
 pydantic[email]
-langchain==0.0.295
-clarifai