Update app.py

app.py

@@ -3,68 +3,62 @@ import gradio as gr
 import requests
 import json
 
+# Retrieve the API token from environment variables
+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.")
+
+# Use the token in 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):
-    api_token = os.getenv("API_TOKEN")
-    if not api_token:
-        return "Error: API token is missing. Please set 'API_TOKEN' in the environment."
-    
-    API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
-    HEADERS = {"Authorization": f"Bearer {api_token}"}
-    
     if not verse.strip():
         return "Please enter a Bible verse."
     
     prompt = f"""
-    <s>[INST] You are JR-Sacred Syntax, an expert biblical scholar specializing in the analysis of Bible verses. 
-    Your task is to analyze the given Bible verse and determine if it contains a figure of speech such as Synecdoche, Metonymy, Hyperbole, Simile, or Paradox. 
-    Follow this structured output format:
+    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:
     {{
-      "verse": "[Provide the NKJV Bible verse]",
+      "verse": "[Provide the Bible verse]",
       "figure": "[Identify the figure of speech]",
-      "explanation": "[Give a clear explanation]",
+      "explanation": "[Clear explanation]",
       "examples": [
         {{
-          "phrase": "[Identify the phrase from the verse]",
-          "analysis": "[Explain why this phrase qualifies as the identified figure of speech]"
+          "phrase": "[Phrase from verse]",
+          "analysis": "[Explanation]"
         }}
       ]
     }}
-    Now, analyze the following Bible verse:
+    Now, analyze this verse:
     "{verse}"
-    [/INST]</s>
     """
     
     payload = {"inputs": prompt}
+    
     try:
         response = requests.post(API_URL, headers=HEADERS, json=payload)
         response.raise_for_status()
         result = response.json()
         
-        if isinstance(result, dict) and "error" in result:
-            return f"API Error: {result['error']}"
-        
-        if isinstance(result, list) and len(result) > 0:
-            generated_text = result[0].get("generated_text", "")
+        if isinstance(result, list) and len(result) > 0 and "generated_text" in result[0]:
+            generated_text = result[0]["generated_text"].strip()
             try:
-                output_json = json.loads(generated_text)
-                return json.dumps(output_json, indent=2)
+                return json.dumps(json.loads(generated_text), indent=2)
             except json.JSONDecodeError:
-                return "Error: Could not parse API response."
+                return "Error: Could not parse API response. The response might not be in the expected JSON format."
         else:
-            return "Error: Unexpected response format."
+            return "Error: Unexpected response format from API."
     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 biblical explanations.",
+    description="Enter a Bible verse to detect and analyze figures of speech with detailed explanations.",
     allow_flagging="never"
 )