Update app.py

app.py

@@ -16,12 +16,12 @@ def format_steps(steps, total_time):
             md_content += f"{content}\n"
             md_content += f"_Thinking time for this step: {thinking_time:.2f} seconds_\n"
             md_content += "\n---\n"
-    if total_time!=0:
+    if total_time != 0:
         md_content += f"\n**Total thinking time: {total_time:.2f} seconds**"
     return md_content
 
-def main(api_key, user_query):
-    if not api_key:
+def main(api_key, user_query, mode):
+    if mode == "private" and not api_key:
         yield "Please enter your Groq API key to proceed."
         return
     
@@ -30,8 +30,11 @@ def main(api_key, user_query):
         return
     
     try:
-        # Initialize the Groq client with the provided API key
-        client = groq.Groq(api_key=api_key)
+        # Initialize the Groq client with the provided API key or the environment variable
+        if mode == "public":
+            client = groq.Groq(api_key=os.getenv("GROQ_API_KEY"))
+        else:
+            client = groq.Groq(api_key=api_key)
     except Exception as e:
         yield f"Failed to initialize Groq client. Error: {str(e)}"
         return
@@ -56,10 +59,12 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column():
+            mode_toggle = gr.Radio(["public", "private"], label="API Key Mode", value="public")
             api_input = gr.Textbox(
                 label="Enter your Groq API Key:",
                 placeholder="Your Groq API Key",
-                type="password"
+                type="password",
+                visible=False  # Initially hidden
             )
             user_input = gr.Textbox(
                 label="Enter your query:",
@@ -73,8 +78,11 @@ with gr.Blocks() as demo:
         with gr.Column():
             output_md = gr.Markdown()
     
-    submit_btn.click(fn=main, inputs=[api_input, user_input], outputs=output_md)
+    # Show/hide the API key input based on the mode toggle
+    mode_toggle.change(lambda mode: gr.update(visible=mode == "private"), mode_toggle, api_input)
+    
+    submit_btn.click(fn=main, inputs=[api_input, user_input, mode_toggle], outputs=output_md)
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()