Update app.py

app.py

@@ -1,43 +1,33 @@
-import os
-
-# Create app.py content (modifying the user's working Colab code slightly for HF Spaces environment variable safety)
-app_content = """import gradio as gr
+import gradio as gr
 import os
 from groq import Groq
+from google.colab import userdata
 
 # Initialize the Groq client
-# On Hugging Face Spaces, set GROQ_API_KEY as a Secret in the Space Settings
-api_key = os.environ.get("GROQ_API_KEY")
-if not api_key:
-    raise ValueError("GROQ_API_KEY environment variable not found. Please add it as a Secret in your Space Settings.")
-
-client = Groq(api_key=api_key)
+# Retrieve API key from Colab's userdata
+os.environ["GROQ_API_KEY"] = userdata.get('ChatBot_API')
+client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
 # Custom CSS to give it a sleek, modern look
-custom_css = \"\"\"
+custom_css = """
 .gradio-container { background-color: #0b0f19; font-family: 'Inter', sans-serif; }
 #title-header { text-align: center; margin-bottom: 20px; }
 #title-header h1 { color: #38bdf8; font-weight: 800; font-size: 2.2rem; }
 .sidebar-panel { background: #111827 !important; border: 1px solid #1f2937 !important; border-radius: 12px !important; }
 .chat-window { border: 1px solid #1f2937 !important; border-radius: 12px !important; background: #111827 !important; }
-\"\"\"
+"""
 
 def chat_stream(message, history, model, system_prompt, temperature, max_tokens):
-    \"\"\"
+    """
     Custom chat function built for gr.Chatbot component format.
-    history is a list of dicts or tuples depending on older/newer Gradio versions.
-    \"\"\"
+    history is a list of dicts: [{"role": "user", "content": "..."}, ...]
+    """
     # 1. Start with the custom system prompt
     messages = [{"role": "system", "content": system_prompt}]
 
     # 2. Append the conversation history
     for turn in history:
-        # Check if history format is dict or list/tuple
-        if isinstance(turn, dict):
-            messages.append({"role": turn["role"], "content": turn["content"]})
-        else:
-            messages.append({"role": "user", "content": turn[0]})
-            messages.append({"role": "assistant", "content": turn[1]})
+        messages.append({"role": turn["role"], "content": turn["content"]})
 
     # 3. Append current user message
     messages.append({"role": "user", "content": message})
@@ -63,9 +53,9 @@ def chat_stream(message, history, model, system_prompt, temperature, max_tokens)
 
 
 # Building the Interactive Interface using Blocks
-with gr.Blocks() as demo:
+with gr.Blocks() as demo: # Removed css and theme from Blocks constructor
 
-    # Title Section
+    # Title Section - Replaced gr.Div with gr.Markdown
     gr.Markdown("# 🚀 Personal AI ChatBot", elem_id="title-header")
     gr.Markdown("A fully customizable, hyper-fast LLM workspace.")
 
@@ -105,7 +95,10 @@ with gr.Blocks() as demo:
 
         # --- RIGHT COLUMN: CHAT INTERFACE ---
         with gr.Column(scale=3):
-            chatbot = gr.Chatbot(elem_classes="chat-window")
+            chatbot = gr.Chatbot(
+                elem_classes="chat-window"
+                # Removed bubble_full_width as it's not a valid argument for gr.Chatbot
+            )
 
             # Textbox and setup for user interaction
             msg_input = gr.Textbox(
@@ -120,17 +113,9 @@ with gr.Blocks() as demo:
                 chatbot=chatbot,
                 textbox=msg_input,
                 additional_inputs=[model_select, system_input, temp_slider, tokens_slider]
+                # Removed type="messages" as it's not a valid argument for gr.ChatInterface
             )
 
 if __name__ == "__main__":
-    demo.launch(css=custom_css, theme=gr.themes.Soft())
-"""
-
-# Write files to disk
-with open("requirements.txt", "w", encoding="utf-8") as f:
-    f.write(requirements_content.strip())
-
-with open("app.py", "w", encoding="utf-8") as f:
-    f.write(app_content.strip())
-
-print("Files created successfully.")
+    # Moved css and theme to launch()
+    demo.launch(css=custom_css, theme=gr.themes.Soft())