Update App.py

App.py

@@ -1,61 +1,66 @@
-import json
+import os
 import gradio as gr
-from transformers import pipeline
+import requests
+import json
 
-# Load the text generation model (can swap for a Bible-specific LLM)
-model_name = "google/flan-t5-large"  # Change if needed
-generator = pipeline("text2text-generation", model=model_name, trust_remote_code=True)
+# Retrieve the API token from secrets
+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.")
 
-def analyze_bible_figure_of_speech(verse):
-    """
-    Analyzes a Bible verse for figures of speech.
-    
-    The function constructs a structured prompt for an LLM, requesting it to:
-      - Identify any figure of speech in the verse (e.g., Metaphor, Hyperbole, Synecdoche, Paradox, Simile).
-      - Extract the phrase demonstrating the figure of speech.
-      - Explain why it's a figure of speech in biblical context.
-    
-    Expected Output:
-      A structured JSON array with:
-      - "figure": Type of figure of speech.
-      - "phrase": The phrase in the verse.
-      - "explanation": Explanation in biblical context.
-    """
+# Use the token in your request headers
+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):
     if not verse.strip():
         return "Please enter a Bible verse."
-
-    prompt = f"""
-    Analyze the following Bible verse and identify any figures of speech present (e.g., Metaphor, Hyperbole, Synecdoche, Paradox, Simile).
-    Provide a JSON response with:
-    - "figure": Type of figure of speech.
-    - "phrase": The phrase in the verse.
-    - "explanation": Why it is a figure of speech in biblical context.
-
-    Bible Verse: \"{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). 
+Return your answer as a JSON array where each element contains:
+  - "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.
+Bible Verse: "{verse}"
+[/INST] Figures:</s>"""
+    
+    payload = {
+        "inputs": prompt,
+    }
     
     try:
-        # Generate the output from the LLM
-        result = generator(prompt, max_length=512, num_return_sequences=1)
-        generated_text = result[0]['generated_text']
+        response = requests.post(API_URL, headers=HEADERS, json=payload)
+        response.raise_for_status()
+        result = response.json()
         
-        # Try parsing the output as JSON
-        try:
-            output_json = json.loads(generated_text)
-            return json.dumps(output_json, indent=2)  # Pretty-print JSON
-        except json.JSONDecodeError:
-            return generated_text  # Return raw text if JSON parsing fails
-    except Exception as e:
-        return f"Error: {str(e)}"
+        if isinstance(result, list) and len(result) > 0:
+            generated_text = result[0].get("generated_text", "")
+            marker = "Figures:"  # Marker to help extract the output
+            if marker in generated_text:
+                generated_text = generated_text.split(marker, 1)[1].strip()
+            try:
+                output_json = json.loads(generated_text)
+                return json.dumps(output_json, indent=2)
+            except json.JSONDecodeError:
+                return generated_text
+        else:
+            return "Error: Unexpected response format."
+    except requests.exceptions.RequestException as e:
+        return f"API Error: {e}"
 
-# Gradio UI
-interface = gr.Interface(
-    fn=analyze_bible_figure_of_speech,
-    inputs=gr.Textbox(label="Enter a Bible Verse", placeholder="E.g., 'I am the vine, you are the branches' - John 15:5", lines=3),
+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)"),
-    title="Bible Figures of Speech Detector",
-    description="Enter a Bible verse to analyze its figures of speech, such as metaphor, simile, or hyperbole. The app will identify the figure, extract the phrase, and provide an explanation."
+    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."
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    demo.launch()