Update app.py

app.py

@@ -1213,7 +1213,7 @@ def fetch_local_news():
     api_key = os.environ['SERP_API']
     url = f'https://serpapi.com/search.json?engine=google_news&q=omaha headline&api_key={api_key}'
     response = requests.get(url)
-    if response.status_code == 200:
+    if response.status_code == 200):
         results = response.json().get("news_results", [])
         news_html = """
         <h2 style="font-family: 'Georgia', serif; color: #ff0000; background-color: #f8f8f8; padding: 10px; border-radius: 10px;">Omaha Today</h2>
@@ -1442,9 +1442,12 @@ def login(username, password):
 
 def toggle_login_signup(show_login):
     if show_login:
-        return gr.update(visible=True), gr.update(visible=False)
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
     else:
-        return gr.update(visible=False), gr.update(visible=True)
+        return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
+
+def close_forms():
+    return gr.update(visible=False), gr.update(visible=False)
 
 def toggle_login_logout(is_logged_in):
     if is_logged_in:
@@ -1457,7 +1460,7 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
     
     def toggle_login_status(is_logged_in):
         is_logged_in = not is_logged_in
-        return gr.update(value=toggle_login_logout(is_logged_in)), is_logged_in
+        return gr.update(value=toggle_login_logout(is_logged_in)), is_logged_in, gr.update(visible=False), gr.update(visible=False)
     
     with gr.Row():
         with gr.Column():
@@ -1511,6 +1514,7 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
                 login_message = gr.Textbox(interactive=False)
                 login_form_button.click(fn=login, inputs=[login_username, login_password], outputs=[login_message])
                 login_button.click(fn=toggle_login_signup, inputs=[gr.State(True)], outputs=[login_form, gr.Column(visible=False)])
+                gr.Button("Close").click(fn=close_forms, inputs=None, outputs=[login_form, signup_form])
             
             signup_form = gr.Column(visible=False)
             with signup_form:
@@ -1522,9 +1526,10 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
                 signup_message = gr.Textbox(interactive=False)
                 signup_form_button.click(fn=signup, inputs=[signup_username, signup_password, signup_password_confirmation], outputs=[signup_message])
                 signup_button.click(fn=toggle_login_signup, inputs=[gr.State(False)], outputs=[signup_form, gr.Column(visible=False)])
+                gr.Button("Close").click(fn=close_forms, inputs=None, outputs=[login_form, signup_form])
     
-    login_form_button.click(fn=toggle_login_status, inputs=[is_logged_in], outputs=[login_button, is_logged_in])
-    login_button.click(fn=toggle_login_status, inputs=[is_logged_in], outputs=[login_button, is_logged_in])
+    login_form_button.click(fn=toggle_login_status, inputs=[is_logged_in], outputs=[login_button, is_logged_in, login_form, signup_form])
+    signup_form_button.click(fn=toggle_login_status, inputs=[is_logged_in], outputs=[login_button, is_logged_in, login_form, signup_form])
 
 demo.queue()
 demo.launch(share=True)