Update app.py

app.py

@@ -51,9 +51,16 @@ def chat_with_llm(query):
 def process_query_with_llm(query):
     prompt = f"User asked: '{query}'. What would be the best search query to use?"
     
-    # Generate response using text_generation without max_length
-    response = client.text_generation(prompt)  # Removed max_length and num_return_sequences
-    return response[0]['generated_text'].strip() if response else "No query generated."
+    # Generate response using text_generation without assuming the structure of the output
+    response = client.text_generation(prompt)
+    
+    # Ensure response is in string format and handle errors
+    if isinstance(response, str):
+        return response.strip()
+    elif isinstance(response, list) and 'generated_text' in response[0]:
+        return response[0]['generated_text'].strip()
+    else:
+        return "No query generated."
 
 # Function to perform a Google search using the googlesearch-python package
 def search_web(query):
@@ -104,6 +111,7 @@ if user_input:
         response = chat_with_llm(user_input)
 
 
+
 # import streamlit as st
 # from huggingface_hub import InferenceClient
 # from googlesearch import search