Update app.py

app.py

@@ -17,10 +17,12 @@ if "creds" not in st.session_state:
     st.session_state.creds = None
 if "auth_url" not in st.session_state:
     st.session_state.auth_url = None
+if "auth_code" not in st.session_state:
+    st.session_state.auth_code = ""
 if "flow" not in st.session_state:
     st.session_state.flow = None
 
-# Authenticate Gmail API
+
 def authenticate_gmail(credentials_file):
     if os.path.exists('token.json'):
         try:
@@ -50,28 +52,26 @@ def authenticate_gmail(credentials_file):
             st.info("Please visit this URL to authorize the application:")
             st.code(st.session_state.auth_url)
 
-            auth_code = st.text_input("Enter the authorization code:")
-            if st.button("Submit Authentication Code"):
-                try:
-                    st.session_state.flow.fetch_token(code=auth_code)
-                    st.session_state.creds = st.session_state.flow.credentials
-                    st.session_state.authenticated = True
-                    with open('token.json', 'w') as token_file:
-                        json.dump({
-                            "token": st.session_state.creds.token,
-                            "refresh_token": st.session_state.creds.refresh_token,
-                            "token_uri": st.session_state.creds.token_uri,
-                            "client_id": st.session_state.creds.client_id,
-                            "client_secret": st.session_state.creds.client_secret,
-                            "scopes": st.session_state.creds.scopes
-                        }, token_file)
-                    st.success("Authentication successful!")
-                except Exception as e:
-                    st.error(f"Error during authentication: {e}")
-
-    return st.session_state.creds
-
-# Fetch Emails from Gmail API
+
+def submit_auth_code():
+    try:
+        st.session_state.flow.fetch_token(code=st.session_state.auth_code)
+        st.session_state.creds = st.session_state.flow.credentials
+        st.session_state.authenticated = True
+        with open('token.json', 'w') as token_file:
+            json.dump({
+                "token": st.session_state.creds.token,
+                "refresh_token": st.session_state.creds.refresh_token,
+                "token_uri": st.session_state.creds.token_uri,
+                "client_id": st.session_state.creds.client_id,
+                "client_secret": st.session_state.creds.client_secret,
+                "scopes": st.session_state.creds.scopes
+            }, token_file)
+        st.success("Authentication successful!")
+    except Exception as e:
+        st.error(f"Error during authentication: {e}")
+
+
 def fetch_emails(service, label):
     emails = []
     results = service.users().messages().list(userId='me', labelIds=[label], maxResults=100).execute()
@@ -98,44 +98,31 @@ def fetch_emails(service, label):
     st.success(f"Fetched {len(emails)} emails from {label}.")
     return emails
 
+
 # Main Page
-def main_page():
-    st.sidebar.title("Steps to Use")
-    st.sidebar.write("""
-    1. Enable Gmail API from [Google Cloud Console](https://console.cloud.google.com/).
-    2. Download the `credentials.json` file.
-    3. Enter your Gmail email and App Password.
-    4. Upload your `credentials.json` file.
-    5. Authenticate and proceed to fetch emails.
-    """)
-
-    st.title("Gmail Email Fetcher")
-
-    user_email = st.text_input("Gmail Email")
-    app_password = st.text_input("App Password", type="password")
-    credentials_file = st.file_uploader("Upload credentials.json", type="json")
-
-    if st.button("Authenticate"):
-        if not user_email or not app_password or not credentials_file:
-            st.error("Please provide all required inputs (Gmail, App Password, credentials.json)")
-        else:
-            with open("credentials.json", "wb") as f:
-                f.write(credentials_file.getbuffer())
-            creds = authenticate_gmail("credentials.json")
-            if creds:
-                service = build('gmail', 'v1', credentials=creds)
-                label = st.selectbox("Select Label", ["INBOX", "SENT", "DRAFTS", "TRASH", "SPAM"])
-                if st.button("Fetch Emails"):
-                    emails = fetch_emails(service, label)
-                    if emails:
-                        df = pd.DataFrame(emails)
-                        st.dataframe(df)
-                        csv = df.to_csv(index=False).encode('utf-8')
-                        st.download_button("Download Emails as CSV", csv, f"{label}_emails.csv", "text/csv")
-
-# Navigation
-if "page" not in st.session_state:
-    st.session_state.page = "main"
-
-if st.session_state.page == "main":
-    main_page()
+st.title("Gmail Email Fetcher")
+
+credentials_file = st.file_uploader("Upload credentials.json", type="json")
+if credentials_file:
+    with open("credentials.json", "wb") as f:
+        f.write(credentials_file.getbuffer())
+    authenticate_gmail("credentials.json")
+
+if st.session_state.auth_url:
+    st.text_input("Enter the authorization code:", key="auth_code")
+    if st.button("Submit Authentication Code"):
+        submit_auth_code()
+
+if st.session_state.authenticated:
+    st.success("You are authenticated!")
+    service = build('gmail', 'v1', credentials=st.session_state.creds)
+    label = st.selectbox("Select Label", ["INBOX", "SENT", "DRAFTS", "TRASH", "SPAM"])
+    if st.button("Fetch Emails"):
+        emails = fetch_emails(service, label)
+        if emails:
+            df = pd.DataFrame(emails)
+            st.dataframe(df)
+            csv = df.to_csv(index=False).encode('utf-8')
+            st.download_button("Download Emails as CSV", csv, f"{label}_emails.csv", "text/csv")
+else:
+    st.warning("You are not authenticated yet.")