Update app.py

app.py

@@ -1,59 +1,78 @@
 import os
 import gradio as gr
+import pytesseract
+import pdfplumber
+import docx
 import requests
-import json
+from PIL import Image
 
-# Retrieve the API token from secrets
+# API Configuration
 api_token = os.getenv("API_TOKEN")
 if not api_token:
-    raise ValueError("API token not found. Make sure 'API_TOKEN' is set in the Secrets.")
+    raise ValueError("API token not found. Set 'API_TOKEN' in your environment variables.")
 
-# Use the token in your request headers
-API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"  # Corrected URL
+API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
 HEADERS = {"Authorization": f"Bearer {api_token}"}
 
-# ... rest of your code ...
+def extract_text_from_pdf(pdf_file):
+    text = ""
+    with pdfplumber.open(pdf_file) as pdf:
+        for page in pdf.pages:
+            text += page.extract_text() + "\n"
+    return text.strip()
 
-def generate_exegesis(passage):
-    if not passage.strip():
-        return "Please enter a Bible passage."
+def extract_text_from_docx(docx_file):
+    doc = docx.Document(docx_file)
+    return "\n".join([para.text for para in doc.paragraphs])
 
-    prompt = f"""<s>[INST] You are a professional Bible Scholar. Provide a detailed exegesis of the following biblical verse, including:
-     The original Greek text and transliteration with word-by-word analysis and meanings, historical and cultural context, and theological significance for:
-
-    {passage} [/INST] Exegesis:</s>"""
-
-    payload = {
-        "inputs": prompt,
-    }
+def extract_text_from_image(image_file):
+    image = Image.open(image_file)
+    return pytesseract.image_to_string(image)
 
+def generate_quiz(text):
+    if not text.strip():
+        return "No content extracted. Please upload a valid file."
+    
+    prompt = f"""<s>[INST] Generate a quiz with multiple-choice questions based on the following content:
+    {text} [/INST] Quiz:</s>"""
+    
+    payload = {"inputs": prompt}
+    
     try:
         response = requests.post(API_URL, headers=HEADERS, json=payload)
         response.raise_for_status()
         result = response.json()
-        print("Full API Response:", json.dumps(result, indent=4))  # Debug output
-
+        
         if isinstance(result, list) and len(result) > 0:
-            generated_text = result[0].get("generated_text", "")
-            marker = "Exegesis:"  # Marker to split on
-            if marker in generated_text:
-                # Return only the text after the marker
-                generated_text = generated_text.split(marker, 1)[1].strip()
-            return generated_text or "Error: No response from model."
+            return result[0].get("generated_text", "Quiz generation failed.")
         else:
             return "Error: Unexpected response format."
     except requests.exceptions.RequestException as e:
         return f"API Error: {e}"
 
+def quiz_app(uploaded_file):
+    file_type = uploaded_file.name.split('.')[-1]
+    extracted_text = ""
+    
+    if file_type == "pdf":
+        extracted_text = extract_text_from_pdf(uploaded_file)
+    elif file_type == "docx":
+        extracted_text = extract_text_from_docx(uploaded_file)
+    elif file_type in ["png", "jpg", "jpeg"]:
+        extracted_text = extract_text_from_image(uploaded_file)
+    
+    if not extracted_text:
+        return "No text extracted. Please upload a valid file."
+    
+    return generate_quiz(extracted_text)
 
-# Gradio interface
-demo = gr.Interface(
-    fn=generate_exegesis,
-    inputs=gr.Textbox(label="Enter Bible Passage", placeholder="e.g., John 3:16"),
-    outputs=gr.Textbox(label="Exegesis Commentary"),
-    title="JR Study Bible",
-    description="Enter a Bible passage to receive insightful exegesis commentary using the Hugging Face API."
+iface = gr.Interface(
+    fn=quiz_app,
+    inputs=gr.File(label="Upload PDF, DOCX, or Image"),
+    outputs=gr.Textbox(label="Generated Quiz"),
+    title="AI Quiz Generator",
+    description="Upload a file to generate a quiz based on its content."
 )
 
-if __name__ == "__main__":  # Corrected this line
-    demo.launch()
+if __name__ == "__main__":
+    iface.launch()