Update app.py

app.py

@@ -2,8 +2,8 @@ import os
 import gradio as gr
 import requests
 import json
-# import logging
 import logging
+
 # Retrieve the API token from secrets
 api_token = os.getenv("API_TOKEN")
 if not api_token:
@@ -13,16 +13,10 @@ if not api_token:
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# 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.")
-
 # 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}"}
 
-
 def generate_exegesis(passage):
     if not passage.strip():
         return "Please enter a Bible passage."
@@ -38,13 +32,12 @@ def generate_exegesis(passage):
         response = requests.post(API_URL, headers=HEADERS, json=payload)
         response.raise_for_status()
         result = response.json()
-        logger.debug("Full API Response: %s", json.dumps(result, indent=4))  # Debug output
+        logger.debug("Full API Response: %s", json.dumps(result, indent=4))
 
         if isinstance(result, list) and len(result) > 0:
             generated_text = result[0].get("generated_text", "")
-            marker = "Exegesis:"  # Marker to split on
+            marker = "Exegesis:"
             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."
         else:
@@ -65,13 +58,12 @@ def ask_any_questions(question):
         response = requests.post(API_URL, headers=HEADERS, json=payload)
         response.raise_for_status()
         result = response.json()
-        logger.debug("Full API Response: %s", json.dumps(result, indent=4))  # Debug output
+        logger.debug("Full API Response: %s", json.dumps(result, indent=4))
 
         if isinstance(result, list) and len(result) > 0:
             answer_text = result[0].get("generated_text", "")
-            marker = "Answer:"  # Marker to split on
+            marker = "Answer:"
             if marker in answer_text:
-                # Return only the text after the marker
                 answer_text = answer_text.split(marker, 1)[1].strip()
             return answer_text or "Error: No response from model."
         else:
@@ -92,13 +84,12 @@ def generate_sermon(topic):
         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
+        logger.debug("Full API Response: %s", json.dumps(result, indent=4))
 
         if isinstance(result, list) and len(result) > 0:
             sermon_text = result[0].get("generated_text", "")
-            marker = "Sermon:"  # Marker to split on
+            marker = "Sermon:"
             if marker in sermon_text:
-                # Return only the text after the marker
                 sermon_text = sermon_text.split(marker, 1)[1].strip()
             return sermon_text or "Error: No response from model."
         else:
@@ -119,13 +110,12 @@ def keyword_search(keyword):
         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
+        logger.debug("Full API Response: %s", json.dumps(result, indent=4))
 
         if isinstance(result, list) and len(result) > 0:
             search_results = result[0].get("generated_text", "")
-            marker = "Search Results:"  # Marker to split on
+            marker = "Search Results:"
             if marker in search_results:
-                # Return only the text after the marker
                 search_results = search_results.split(marker, 1)[1].strip()
             return search_results or "Error: No response from model."
         else:
@@ -173,4 +163,4 @@ bible_app = gr.TabbedInterface(
 )
 
 if __name__ == "__main__":
-    bible_app.launch()
+    bible_app.launch()