Spaces:
Runtime error
Runtime error
| 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() |