Update app.py

app.py

@@ -3,12 +3,12 @@ import gradio as gr
 import requests
 import json
 
-# Retrieve the API token from environment variables
+# 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 environment variables.")
+    raise ValueError("API token not found. Make sure 'API_TOKEN' is set in the Secrets.")
 
-# Use the token in request headers
+# 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}"}
 
@@ -16,51 +16,58 @@ def analyze_figure_of_speech(verse):
     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"""
-    You are JR-Sacred Syntax, an expert biblical scholar. Analyze the given Bible verse and determine if it contains a figure of speech such as Synecdoche, Metonymy, Hyperbole, Simile, or Paradox. 
-    Return the response in JSON format with the following structure:
+    [INST] You are JR-Sacred Syntax, an expert biblical scholar specializing in Bible figures of speech.
+    Identify the literary device in the following verse (e.g., Synecdoche, Metonymy, Hyperbole, Simile, Paradox) and structure the response as JSON:
+    
     {{
-      "verse": "[Provide the Bible verse]",
+      "verse": "{verse}",
       "figure": "[Identify the figure of speech]",
-      "explanation": "[Clear explanation]",
+      "explanation": "[Give a detailed yet concise explanation]",
       "examples": [
         {{
-          "phrase": "[Phrase from verse]",
-          "analysis": "[Explanation]"
+          "phrase": "[Highlight the specific phrase]",
+          "analysis": "[Explain why this phrase fits the identified figure of speech]"
         }}
       ]
     }}
-    Now, analyze this verse:
-    "{verse}"
+    [/INST]
     """
     
-    payload = {"inputs": prompt}
+    payload = {"inputs": prompt, "parameters": {"max_new_tokens": 250, "temperature": 0.7, "top_p": 0.9}}
     
     try:
         response = requests.post(API_URL, headers=HEADERS, json=payload)
         response.raise_for_status()
         result = response.json()
         
-        if isinstance(result, list) and len(result) > 0 and "generated_text" in result[0]:
-            generated_text = result[0]["generated_text"].strip()
+        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:
-                return json.dumps(json.loads(generated_text), indent=2)
+                output_json = json.loads(generated_text)
+                return json.dumps(output_json, indent=2)
             except json.JSONDecodeError:
-                return "Error: Could not parse API response. The response might not be in the expected JSON format."
+                return generated_text
         else:
-            return "Error: Unexpected response format from API."
+            return "Error: Unexpected response format."
     except requests.exceptions.RequestException as e:
         return f"API Error: {e}"
 
-# Define Gradio UI
 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),
+    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="JR-Sacred Syntax: Bible Figures of Speech Detector",
-    description="Enter a Bible verse to detect and analyze figures of speech with detailed explanations.",
-    allow_flagging="never"
+    description="Enter a Bible verse to detect and analyze figures of speech with detailed biblical explanations."
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()