Update app.py

app.py

@@ -1,7 +1,5 @@
 import os
 import time
-import json
-import base64
 from datetime import datetime
 import gradio as gr
 from google import genai
@@ -36,10 +34,10 @@ def get_greeting():
     elif h < 17: return "Good afternoon 👋"
     else: return "Good evening 👋"
 
-# ── Gemini Stream & Logic Functions ──
+# ── Gemini Stream Logic ──
 def predict(message, history, subject_id, file_input):
     if not client:
-        yield "⚠️ Please configure GEMINI_API_KEY in your Hugging Face Space secrets."
+        yield "⚠️ Please add GEMINI_API_KEY in Space Settings -> Secrets."
         return
 
     subj = next((s["label"] for s in SUBJECTS if s["id"] == subject_id), "General Studies")
@@ -47,12 +45,12 @@ def predict(message, history, subject_id, file_input):
     
     formatted_contents = []
     if history:
-        for turn in history:
-            if isinstance(turn, dict):
-                formatted_contents.append(types.Content(role=turn.get("role", "user"), parts=[types.Part.from_text(str(turn.get("content", "")))]))
-            elif isinstance(turn, (list, tuple)) and len(turn) >= 2:
-                if turn[0]: formatted_contents.append(types.Content(role="user", parts=[types.Part.from_text(str(turn[0]))]))
-                if turn[1]: formatted_contents.append(types.Content(role="model", parts=[types.Part.from_text(str(turn[1]))]))
+        for msg in history:
+            if isinstance(msg, dict) and "role" in msg and "content" in msg:
+                role = "user" if msg["role"] == "user" else "model"
+                text = msg["content"]
+                if text:
+                    formatted_contents.append(types.Content(role=role, parts=[types.Part.from_text(str(text))]))
 
     current_parts = [types.Part.from_text(message)]
     
@@ -91,36 +89,36 @@ def predict(message, history, subject_id, file_input):
     except Exception as e:
         yield f"An error occurred: {str(e)}"
 
-# ── Dynamic Features Generators ──
+# ── Generators ──
 def generate_quiz(subject_id):
-    if not client: return "API key missing."
+    if not client: return "⚠️ Please add GEMINI_API_KEY in Space Settings."
     subj = next((s["label"] for s in SUBJECTS if s["id"] == subject_id), "General Studies")
     prompt = f"Generate a 3-question multiple choice quiz about {subj} with answers at the bottom. Format beautifully with markdown."
     try:
         response = client.models.generate_content(model='gemini-2.5-flash', contents=prompt)
         return response.text
     except Exception as e:
-        return f"Error generating quiz: {str(e)}"
+        return f"Error: {str(e)}"
 
 def generate_notes(history, subject_id):
-    if not client: return "API key missing."
+    if not client: return "⚠️ Please add GEMINI_API_KEY in Space Settings."
     subj = next((s["label"] for s in SUBJECTS if s["id"] == subject_id), "General Studies")
-    prompt = f"Based on this study history for {subj}: {str(history)}. Create highly structured, clean summary revision notes with key bullet points."
+    prompt = f"Based on this study history for {subj}: {str(history)}. Create structured summary revision notes."
     try:
         response = client.models.generate_content(model='gemini-2.5-flash', contents=prompt)
         return response.text
     except Exception as e:
-        return f"Error generating notes: {str(e)}"
+        return f"Error: {str(e)}"
 
 def generate_flashcards(history, subject_id):
-    if not client: return "API key missing."
+    if not client: return "⚠️ Please add GEMINI_API_KEY in Space Settings."
     subj = next((s["label"] for s in SUBJECTS if s["id"] == subject_id), "General Studies")
-    prompt = f"Based on this study history for {subj}: {str(history)}. Generate 4 interactive flashcards formatted as 'Question: [Text] | Answer: [Text]'."
+    prompt = f"Based on this study history for {subj}: {str(history)}. Generate 4 interactive flashcards."
     try:
         response = client.models.generate_content(model='gemini-2.5-flash', contents=prompt)
         return response.text
     except Exception as e:
-        return f"Error generating flashcards: {str(e)}"
+        return f"Error: {str(e)}"
 
 # ── Gradio UI Construction ──
 with gr.Blocks() as demo:
@@ -130,7 +128,7 @@ with gr.Blocks() as demo:
         # SIDEBAR
         with gr.Column(scale=1, min_width=280):
             gr.Markdown("## ⚡ StudyAI Dashboard")
-            greeting_md = gr.Markdown(f"*{get_greeting()}*")
+            gr.Markdown(f"*{get_greeting()}*")
             
             subject_dropdown = gr.Dropdown(
                 choices=[(s["label"] + " " + s["emoji"], s["id"]) for s in SUBJECTS],
@@ -141,10 +139,10 @@ with gr.Blocks() as demo:
             
             gr.Markdown("---")
             gr.Markdown("⏱️ **Focus Pomodoro Timer**")
-            gr.HTML("<div style='font-size:24px; font-weight:bold; text-align:center;'>25:00</div>")
+            pomo_display = gr.HTML("<div style='font-size:24px; font-weight:bold; text-align:center;'>25:00</div>")
             with gr.Row():
-                gr.Button("▶️ Start", size="sm")
-                gr.Button("⏹️ Reset", size="sm")
+                start_btn = gr.Button("▶️ Start", size="sm")
+                reset_btn = gr.Button("⏹️ Reset", size="sm")
             
             gr.Markdown("---")
             gr.Markdown("📝 **Todo Checklist**")
@@ -155,7 +153,7 @@ with gr.Blocks() as demo:
             new_todo = gr.Textbox(placeholder="+ Add custom task & press Enter", show_label=False)
             
             def add_todo(task, current_todos):
-                if task and task.strip(): 
+                if task and task.strip():
                     current_todos.append(task.strip())
                 return gr.update(choices=current_todos, value=current_todos), ""
             new_todo.submit(add_todo, [new_todo, todo_list], [todo_list, new_todo])
@@ -170,16 +168,15 @@ with gr.Blocks() as demo:
                     
                     suggestion_box = gr.Radio(
                         choices=SUGGESTIONS["math"], 
-                        label="💡 Instant Suggestions (Click to choose, then type or send)",
+                        label="💡 Instant Suggestions",
                         interactive=True
                     )
                     
-                    file_upload = gr.File(label="📂 Upload Study Material (PDF up to 50MB)", file_types=[".pdf"])
-                    chatbot = gr.Chatbot(height=400)
+                    file_upload = gr.File(label="📂 Upload Study Material (PDF)", file_types=[".pdf"])
+                    chatbot = gr.Chatbot(height=400, type="messages")
                     
                     with gr.Row():
                         msg_input = gr.Textbox(placeholder="Ask anything...", scale=4, show_label=False)
-                        audio_input = gr.Audio(sources=["microphone"], type="filepath", label="🎙️ Voice", scale=1)
                         submit_btn = gr.Button("Send 🚀", variant="primary", scale=1)
 
                     def update_subject_view(sub_id):
@@ -194,28 +191,20 @@ with gr.Blocks() as demo:
                         outputs=[subject_info_banner, suggestion_box, current_subject_id, chatbot]
                     )
                     
-                    suggestion_box.select(lambda x: x.value, inputs=[suggestion_box], outputs=[msg_input])
-
-                    def handle_audio(audio_path):
-                        if audio_path: 
-                            return "[Voice Note Uploaded: Analyzing audio content]"
-                        return ""
-                    audio_input.change(handle_audio, inputs=[audio_input], outputs=[msg_input])
+                    suggestion_box.select(lambda evt: evt.value, outputs=[msg_input])
 
                     def respond(message, chat_history, sub_id, pdf_file):
                         if not message or not message.strip(): 
-                            return "", chat_history
+                            yield "", chat_history
+                            return
                         
-                        chat_history = chat_history or []
-                        chat_history.append({"role": "user", "metadata": {"title": ""}, "content": message})
-                        chat_history.append({"role": "assistant", "metadata": {"title": ""}, "content": ""})
+                        chat_history = list(chat_history or [])
+                        chat_history.append({"role": "user", "content": message})
+                        chat_history.append({"role": "assistant", "content": ""})
                         
-                        raw_history = []
-                        for i in range(0, len(chat_history)-2, 2):
-                            if i+1 < len(chat_history):
-                                raw_history.append((chat_history[i]["content"], chat_history[i+1]["content"]))
+                        history_for_api = chat_history[:-2]
 
-                        for chunk in predict(message, raw_history, sub_id, pdf_file):
+                        for chunk in predict(message, history_for_api, sub_id, pdf_file):
                             chat_history[-1]["content"] = chunk
                             yield "", chat_history