Update app.py

app.py

@@ -62,16 +62,16 @@ class SentenceTransformerEmbeddings:
             print(f"Error loading SentenceTransformer: {e}")
             raise
 
-    def __call__(self, texts):
-        if isinstance(texts, str):
-            texts = [texts]
+    def __call__(self, input):
+        if isinstance(input, str):
+            input = [input]
         try:
-            embeddings = self.model.encode(texts, convert_to_numpy=True).tolist()
+            embeddings = self.model.encode(input, convert_to_numpy=True).tolist()
             return embeddings
         except Exception as e:
             print(f"Error generating embeddings: {e}")
             # Fallback to simple embeddings if needed
-            return [[0.0] * 384 for _ in texts]
+            return [[0.0] * 384 for _ in input]
 
 # Initialize ChromaDB with a persistent directory in a writable location
 persistent_dir = os.path.join("/tmp", "chroma_db")
@@ -285,7 +285,6 @@ def ask_question():
          print(f"Error in /ask_question endpoint: {e}")
          return jsonify({"response": "Sorry, an error occurred while generating the response."}), 500
 
-
 if __name__ == '__main__':
     # Make sure GOOGLE_API_KEY is checked before starting the app
     if not GOOGLE_API_KEY:
@@ -293,4 +292,5 @@ if __name__ == '__main__':
     else:
          # Use port 7860 for Hugging Face Spaces compatibility
          port = int(os.environ.get("PORT", 7860))
-         app.run(debug=False, host='0.0.0.0', port=port)
+         app.run(debug=False, host='0.0.0.0', port=port)
+