Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import RagTokenizer, RagRetriever, RagTokenForGeneration
 import json
+import numpy as np
 
 # Load horoscope data
 with open("horoscope_data.json", "r") as file:
@@ -12,13 +13,22 @@ class CustomHoroscopeRetriever(RagRetriever):
         self.horoscope_data = horoscope_data
 
     def retrieve(self, question_texts, n_docs=1):
+        # Convert numpy arrays to lists if needed
+        if isinstance(question_texts, np.ndarray):
+            question_texts = question_texts.tolist()
+        
         # Ensure question_texts is a list of strings
         if isinstance(question_texts, list):
             question_texts = question_texts[0]  # Get the first element
         if isinstance(question_texts, list):  # If it's still a list, get the first string
             question_texts = question_texts[0]
         
-        zodiac_sign = question_texts.capitalize()  # Now it should be a string
+        # Ensure the text is a string
+        if isinstance(question_texts, str):
+            zodiac_sign = question_texts  # Use as-is
+        else:
+            return ["I couldn't process your request. Please try again with a valid zodiac sign."]
+        
         if zodiac_sign in self.horoscope_data:
             return [self.horoscope_data[zodiac_sign]]
         else: