Streamlit frontend

frontend/Login.py

@@ -0,0 +1,59 @@
+import json
+import requests
+
+import streamlit as st
+        
+
+def auth_page():
+
+    base_url = 'http://127.0.0.1:8000'
+
+
+    headers={"accept":"application/json"}
+
+    tab1, tab2 = st.tabs(["Signup", "Login"])
+
+    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")
+            login_button = st.form_submit_button(label="Login")
+
+        with st.spinner("Logging in..."):
+            if login_button:
+                try:
+                    credentials = {"username":username, "password":password}
+                    response = requests.post(base_url + "/auth/login", headers=headers, data=json.dumps(credentials)) 
+                    if (response.status_code!=200):
+                        raise Exception("Login Failed")
+                    # res_dict.update(response.json())
+                    st.session_state["username"] = username
+                    st.session_state["access_token"] = response.json()['access_token']
+                    st.session_state["refresh_token"] = response.json()['refresh_token']
+                    st.success("Logged in successfully")
+                    st.experimental_rerun()
+
+                except Exception as e:
+                    st.error(e)
+
+

frontend/__init__.py

@@ -0,0 +1 @@
+    

frontend/pages/Code.py

@@ -0,0 +1,88 @@
+import json
+import requests
+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",
+    page_icon="👋",
+    layout="wide",
+    initial_sidebar_state="expanded",
+)
+
+st.write("# DocGup-tea: AI based Documentation Generator 💀")
+
+def logout():
+    del st.session_state["access_token"]
+    del st.session_state["refresh_token"]
+    del st.session_state["username"]
+
+with st.sidebar:
+    if 'username' not in st.session_state:
+        st.header("Login/Signup")
+    else:
+        st.header(f"Welcome, {st.session_state.username}!")
+        st.warning("Generating a new API Key will invalidate the previous one from all your projects. Do you wish to continue?")
+        if st.button("Generate API KEY"):
+            with st.spinner("Generating API Key..."):
+                try:
+                    base_url = "http://localhost:8000"
+                    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):
+                        raise Exception("API Key Generation Failed")
+                    st.info("Please save the API KEY as it will be shown only once.")
+                    st.code(response.json()["api_key"],"bash")
+                    st.success("API Key Generated Successfully")
+                except Exception as e:
+                    st.error(e)
+        
+
+
+        with st.expander("More Options"):
+            if st.button("Logout"):
+                logout()
+                st.experimental_rerun()
+
+
+def code_page():
+    base_url = 'http://localhost:8000'
+
+    def query_post(url, headers, data=None, params=None):
+        response = requests.post(url, data=data, headers=headers, params=params)
+        return response
+    
+    headers={"accept":"application/json"}
+
+
+    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.")
+
+    code_input = st.text_area("Code Input", height=200)
+    comment_placeholder = st.empty()
+
+    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})
+            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
+            comment_placeholder.empty()
+            comment_placeholder.markdown(f"<pre><code>{docstr}</code></pre>", unsafe_allow_html=True)
+        else:
+            st.warning("Please enter some code.")
+
+
+if 'access_token' not in st.session_state:
+    st.session_state.runpage = auth_page  
+else:
+    st.session_state.runpage = code_page
+st.session_state.runpage()

frontend/🏡 Home.py

@@ -0,0 +1,90 @@
+import streamlit as st
+from Login import auth_page
+from PIL import Image 
+
+import base64
+
+# image2=Image.open('assets/logo2.png')
+st.set_page_config(
+    page_title="DocGup-tea",
+    layout="wide",
+    page_icon="🏡",
+    initial_sidebar_state="expanded",
+)
+
+@st.cache_data
+def get_base64_bin_file(bin_file):
+    with open(bin_file, 'rb') as f:
+        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 📃")
+
+def logout():
+    del st.session_state["access_token"]
+    del st.session_state["refresh_token"]
+    del st.session_state["username"]
+
+with st.sidebar:
+    if 'username' not in st.session_state:
+        st.header("Login/Signup")
+    else:
+        st.header(f"Welcome, {st.session_state.username}!")
+        if st.button("Logout"):
+            logout()
+            st.experimental_rerun()
+
+
+def home_page():
+
+
+    def set_page_background(png_file):
+        bin_str = get_base64_bin_file(png_file)
+        page_bg_img = f'''
+            <style>
+            .stApp {{
+                background-image: url("data:image/jpg;base64,{bin_str}");
+                background-size: cover;
+                background-repeat: no-repeat;
+                background-attachment: scroll;
+            }}
+            </style>
+        '''
+        st.markdown(page_bg_img, unsafe_allow_html=True)
+
+    set_page_background("../assets/bg.jpg") 
+
+    st.markdown(
+        """
+
+        ## A step-by-step guide 
+
+        The process is quite straightforward. BOB offers loans to eligible applicants with strong financial profiles. 
+        Individuals need to provide their basic personal, employment, income and property details to know if you are fit to apply for a loan.
+
+        ### 1 . Login 
+
+        Step one is the login part, you just have to work your way through the following simple steps.
+        - Enter your mobile number
+        - Enter the OTP recieved 
+        - accept consent in the Safe portal
+
+        ### 2. Loan page
+
+        Once you login, you will be redirected to the loan page.
+        - Offer all relevant details such as loan amount, loan history, income, etc.
+        - Click on the submit option once you have filled in all the details.
+        - Our algorithm will assess your eligibility based on the details provided by you and you will be awarded with a `yes` or a `no`.
+    """
+    )
+
+if 'access_token' not in st.session_state:
+    st.session_state.runpage = auth_page
+else:
+    st.session_state.runpage = home_page
+st.session_state.runpage()
+