Update app.py

app.py

@@ -3,6 +3,11 @@ from huggingface_hub import InferenceClient
 from textblob import TextBlob
 import json
 import os
+import time
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
 
 # Get the API token from the environment variable
 api_token = os.getenv('HUGGINGFACEHUB_API_TOKEN')
@@ -70,12 +75,23 @@ def respond(message, history, system_message, max_tokens, temperature, top_p, fe
 
     formatted_prompt = format_alpaca_prompt(message, system_message, history)
 
-    response = client.text_generation(
-        formatted_prompt,
-        max_new_tokens=max_tokens,
-        temperature=temperature,
-        top_p=top_p,
-    )
+    # Retry mechanism
+    max_retries = 3
+    for attempt in range(max_retries):
+        try:
+            response = client.text_generation(
+                formatted_prompt,
+                max_new_tokens=max_tokens,
+                temperature=temperature,
+                top_p=top_p,
+            )
+            break  # Exit the loop if the request is successful
+        except Exception as e:
+            logging.error(f"Attempt {attempt + 1} failed: {e}")
+            if attempt < max_retries - 1:
+                time.sleep(2 ** attempt)  # Exponential backoff
+            else:
+                raise e
 
     # ✅ Extract only the response
     cleaned_response = response.split("### Response:")[-1].strip()
@@ -90,6 +106,22 @@ def collect_feedback(response, feedback):
     """Collect user feedback on the chatbot's response"""
     save_interaction(response, feedback=feedback)
 
+def view_interactions():
+    if os.path.exists(INTERACTIONS_FILE):
+        with open(INTERACTIONS_FILE, "r") as file:
+            interactions = json.load(file)
+            return json.dumps(interactions, indent=4)
+    else:
+        return "No interactions found."
+
+# Create a Gradio interface to display interactions
+view_interface = gr.Interface(
+    fn=view_interactions,
+    inputs=[],
+    outputs="text",
+    title="View Interactions"
+)
+
 feedback_interface = gr.Interface(
     fn=collect_feedback,
     inputs=[
@@ -112,4 +144,5 @@ demo = gr.ChatInterface(
 
 if __name__ == "__main__":
     demo.launch()
-    feedback_interface.launch()
+    feedback_interface.launch()
+    view_interface.launch()