Update app.py

app.py

@@ -2,80 +2,80 @@
 # UNEB Primary 7/6 Exam Preparation Platform
 # Multi-AI System: Gemini (Primary) → Cohere → HF (Fallback)
 # ========================================================
-
 import os
 import json
-                    with gr.Tabs():
-                        with gr.Tab("✏️ Draw Answers"):
-                            canvas = gr.Sketchpad(
-                                label="Draw or write your answers here",
-                                type="pil",
-                                height=500,
-                                brush=gr.Brush(
-                                    colors=["#000000", "#0000FF", "#FF0000"],
-                                    default_size=4
-                                )
-                            )
-                            canvas_status = gr.Textbox(
-                                label="Canvas Status",
-                                interactive=False,
-                                value="Ready to draw"
-                            )
-
-                        with gr.Tab("📸 Upload Photo"):
-                            upload_image = gr.Image(
-                                label="Upload photo of your written work",
-                                type="pil",
-                                height=500
-                            )
-                            upload_status = gr.Textbox(
-                                label="Upload Status",
-                                interactive=False,
-                                value="Ready to upload"
-                            )
-
-                        with gr.Tab(" Type Answers"):
-                            gr.Markdown("""
-### 📝 Type Your Answers
-**Instructions:** Type each answer in the box below. Use one of these formats:
-- **Option A (Recommended):** Type answers separated by blank lines
-  ```
-  Answer to question 1 goes here
-  
-  Answer to question 2 goes here
-  
-  Answer to question 3 goes here
-  ```
-- **Option B:** Type with labels
-  ```
-  Answer 1: Your answer here
-  Answer 2: Your answer here
-  Answer 3: Your answer here
-  ```
-- **Option C:** Type with Q labels
-  ```
-  Q1: Your answer here
-  Q2: Your answer here
-  Q3: Your answer here
-  ```
-                            """)
-
-                            typed_answers = gr.Textbox(
-                                label="Type Your Answers Here",
-                                lines=20,
-                                placeholder="Type your answers using any of the formats shown above...",
-                                elem_classes="answer-input"
-                            )
+from datetime import datetime
+import gradio as gr
+import time
+from io import BytesIO
+from PIL import Image
+import csv
+from pathlib import Path
+import re
+
+# ---------- 1. Configure AI Systems ----------
+try:
+    import google.generativeai as genai
+    genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+    gemini_model = genai.GenerativeModel('gemini-2.0-flash-exp')
+    print("✓ Gemini AI initialized (PRIMARY with Vision)")
+except Exception as e:
+    print(f"✗ Gemini Error: {e}")
+    gemini_model = None
+
+try:
+    import cohere
+    cohere_client = cohere.Client(os.getenv("COHERE_API_KEY"))
+    print("✓ Cohere initialized (SECONDARY)")
+except Exception as e:
+    print(f"✗ Cohere Error: {e}")
+    cohere_client = None
+
+try:
+    from huggingface_hub import InferenceClient
+    hf_client = InferenceClient(api_key=os.environ.get("HF_TOKEN"))
+    print("✓ Hugging Face initialized (FALLBACK)")
+except Exception as e:
+    print(f"✗ HF Error: {e}")
+    hf_client = None
+
+
+def ask_ai(prompt, temperature=0.7, max_retries=2, image=None):
+    """Try models: Gemini → Cohere → HF
+    If image provided, only Gemini can process it"""
+
+    # Try Gemini first (Primary) - handles both text and images
+    if gemini_model:
+        for attempt in range(max_retries):
+            try:
+                if image:
+                    response = gemini_model.generate_content(
+                        [prompt, image],
+                        generation_config=genai.types.GenerationConfig(
+                            temperature=temperature,
+                        )
+                    )
+                else:
+                    response = gemini_model.generate_content(
+                        prompt,
+                        generation_config=genai.types.GenerationConfig(
+                            temperature=temperature,
+                        )
+                    )
+                return response.text, "gemini"
+            except Exception as e:
+                print(f"✗ Gemini attempt {attempt+1} failed: {e}")
+                if attempt < max_retries - 1:
+                    time.sleep(1)
 
-                    # Combined submit button (outside tabs, still inside right column)
-                    with gr.Row():
-                        submit_btn = gr.Button(" Submit for Correction", variant="primary", size="lg")
+    if image:
+        return " Image analysis requires Gemini AI. Please check API configuration.", "error"
 
-                    submit_status = gr.Textbox(
-                        label="Submission Status",
-                        interactive=False,
-                        lines=2
-                    )
+    # Try Cohere (Secondary) - text only
+    if cohere_client:
+        for attempt in range(max_retries):
+            try:
+                response = cohere_client.chat(
                     model="command-r-plus-08-2024",
                     message=prompt,
                     temperature=temperature
@@ -85,7 +85,7 @@ import json
                 print(f"✗ Cohere attempt {attempt+1} failed: {e}")
                 if attempt < max_retries - 1:
                     time.sleep(1)
-    
+
     # Try Hugging Face (Fallback) - text only
     if hf_client:
         try:
@@ -98,7 +98,7 @@ import json
             return completion.choices[0].message.content, "hf"
         except Exception as e:
             print(f"✗ HF failed: {e}")
-    
+
     return " All AI services failed. Please try again later.", "error"
 
 # ---------- 3. Uganda Primary Curriculum Syllabus (P6 & P7 Only, by Subject) ----------