Update app.py

app.py

@@ -1,35 +1,38 @@
 import os
+import json
+import tempfile
 import gradio as gr
 import requests
-import json
+from docx import Document
 
-# Retrieve the API token from secrets
+# Retrieve the API token from the environment
 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. Please set the API_TOKEN environment variable.")
 
-# Use the token in your request headers
+# Define the API URL and headers for the Hugging Face Inference API
 API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
 HEADERS = {"Authorization": f"Bearer {api_token}"}
 
 def analyze_figure_of_speech(verse):
+    """
+    Send the Bible verse to the LLM for analysis of figures of speech.
+    The prompt instructs the model to analyze the verse and return a JSON array.
+    """
     if not verse.strip():
         return "Please enter a Bible verse."
     
-    # Construct a prompt to instruct the model to analyze figures of speech in the verse.
-    prompt = f"""<s>[INST] You are JR-Sacred Syntax, an expert biblical scholar specializing in the analysis of Bible verses. 
-Analyze the following Bible verse to identify any figures of speech (e.g., Metaphor, Synecdoche, Hyperbole, Simile, Paradox and others). 
-Return your answer as a JSON array where each element contains:
-  - VERSES: show the verse in NKJV,
-  - FIGURE: the type of figure of speech,
-  - PHRASE: the exact phrase in the verse,
-  - EXPLANATION: a brief explanation of why it qualifies as that figure in a biblical context.
+    prompt = f"""<s>[INST] You are JR-Sacred Syntax, an expert biblical scholar specializing in Bible analysis.
+Analyze the following Bible verse to identify any figures of speech (e.g., Metaphor, Synecdoche, Hyperbole, Simile, Paradox).
+Return your answer as a JSON array where each element includes:
+  - "verses": the Bible verse reference,
+  - "figure": the type of figure of speech,
+  - "phrase": the exact phrase from the verse,
+  - "explanation": a brief explanation in biblical context.
 Bible Verse: "{verse}"
 [/INST] Figures:</s>"""
     
-    payload = {
-        "inputs": prompt,
-    }
+    payload = {"inputs": prompt}
     
     try:
         response = requests.post(API_URL, headers=HEADERS, json=payload)
@@ -38,7 +41,7 @@ Bible Verse: "{verse}"
         
         if isinstance(result, list) and len(result) > 0:
             generated_text = result[0].get("generated_text", "")
-            marker = "Figures:"  # Marker to help extract the output
+            marker = "Figures:"
             if marker in generated_text:
                 generated_text = generated_text.split(marker, 1)[1].strip()
             try:
@@ -51,17 +54,61 @@ Bible Verse: "{verse}"
     except requests.exceptions.RequestException as e:
         return f"API Error: {e}"
 
-demo = gr.Interface(
-    fn=analyze_figure_of_speech,
-    inputs=gr.Textbox(
-        label="Enter Bible Verse", 
-        placeholder="e.g., 'I am the vine, you are the branches' - John 15:5", 
-        lines=3
-    ),
-    outputs=gr.Textbox(label="Figures of Speech Analysis (JSON)"),
+def generate_docx(analysis_json):
+    """
+    Parse the analysis JSON and create a formatted DOCX file.
+    Each entry is added with headings and bullet points.
+    Returns the file path of the generated DOCX.
+    """
+    try:
+        analysis_data = json.loads(analysis_json)
+    except json.JSONDecodeError:
+        # If not valid JSON, treat the entire text as one entry.
+        analysis_data = [{"verses": "", "figure": "", "phrase": "", "explanation": analysis_json}]
+    
+    document = Document()
+    document.add_heading("Bible Figures of Speech Analysis", level=1)
+    
+    for item in analysis_data:
+        verses = item.get("verses", "Unknown Verse")
+        figure = item.get("figure", "Unknown Figure")
+        phrase = item.get("phrase", "Unknown Phrase")
+        explanation = item.get("explanation", "No explanation provided.")
+        
+        document.add_heading(f"Verse: {verses}", level=2)
+        document.add_paragraph(f"Figure: {figure}", style='List Bullet')
+        document.add_paragraph(f"Phrase: {phrase}", style='List Bullet')
+        document.add_paragraph(f"Explanation: {explanation}", style='List Bullet')
+        document.add_paragraph("")  # Add an empty paragraph for spacing
+
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".docx")
+    document.save(temp_file.name)
+    return temp_file.name
+
+def process_and_download(verse):
+    """
+    Process the Bible verse:
+      1. Get the analysis result as JSON.
+      2. Generate a formatted DOCX file from the analysis.
+    Returns both the JSON result (as text) and the DOCX file path.
+    """
+    analysis = analyze_figure_of_speech(verse)
+    docx_file = generate_docx(analysis)
+    return analysis, docx_file
+
+# Create the Gradio Interface with two outputs: JSON analysis and a DOCX download.
+iface = gr.Interface(
+    fn=process_and_download,
+    inputs=gr.Textbox(label="Enter Bible Verse", 
+                      placeholder="E.g., 'I am the vine, you are the branches' - John 15:5", 
+                      lines=3),
+    outputs=[
+        gr.Textbox(label="Figures of Speech Analysis (JSON)"),
+        gr.File(label="Download Analysis (.docx)")
+    ],
     title="JR-Sacred Syntax: Bible Figures of Speech Detector",
-    description="Enter a Bible verse to detect and analyze figures of speech with detailed biblical explanations."
+    description="Enter a Bible verse to detect and analyze figures of speech with detailed biblical explanations. Download the formatted analysis as a DOCX file."
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    iface.launch()