Update app.py

app.py

@@ -15,9 +15,11 @@ async def generate_solution_python(user_query):
 
     Args:
         user_query (str): The query provided by the user.
+    Returns:
+        str: The generated solution text or an error message.
     """
     if not user_query:
-        return "Please enter your query to get a solution."
+        return "Error: Query is required."
 
     print(f"Processing query: {user_query}")
     response_text = ""
@@ -32,6 +34,8 @@ async def generate_solution_python(user_query):
         
         # In a real scenario, you'd make an API call like this (example with a hypothetical search API):
         # search_api_key = os.environ.get("YOUR_SEARCH_API_KEY")
+        # if not search_api_key:
+        #     raise ValueError("YOUR_SEARCH_API_KEY environment variable not set.")
         # search_api_url = "https://api.example.com/search"
         # search_response = requests.get(search_api_url, params={"q": user_query, "api_key": search_api_key})
         # search_response.raise_for_status()
@@ -85,17 +89,17 @@ Please provide a detailed and helpful solution, incorporating the provided infor
             response_text = "No solution could be generated. Please try a different query."
 
     except requests.exceptions.RequestException as e:
-        error_message = f"Network or API error: {e}"
+        error_message = f"Network or API error during LLM call: {e}"
         print(f"Error: {error_message}")
-        response_text = f"An error occurred: {error_message}. Please check the logs for details."
+        response_text = f"An API error occurred: {error_message}. Please check the logs."
     except ValueError as e:
-        error_message = f"Configuration error: {e}"
+        error_message = f"Configuration error (e.g., missing API key): {e}"
         print(f"Error: {error_message}")
-        response_text = f"An error occurred: {error_message}. Please check the logs for details."
+        response_text = f"A configuration error occurred: {error_message}. Please check your Space secrets."
     except Exception as e:
-        error_message = f"An unexpected error occurred: {e}"
+        error_message = f"An unexpected error occurred in generate_solution_python: {e}"
         print(f"Error: {error_message}")
-        response_text = f"An error occurred: {error_message}. Please check the logs for details."
+        response_text = f"An unexpected error occurred: {error_message}. Please check the logs."
     
     return response_text
 
@@ -109,14 +113,24 @@ def index():
 @app.route('/generate', methods=['POST'])
 async def generate():
     """Handles the AI generation request."""
-    data = request.json
-    user_query = data.get('query')
-    if not user_query:
-        return jsonify({"error": "Query is required"}), 400
+    try:
+        # Try to parse JSON from the request body
+        data = request.get_json()
+        if not data:
+            return jsonify({"error": "Request body must be JSON"}), 400
+
+        user_query = data.get('query')
+        if not user_query:
+            return jsonify({"error": "Query is required in the request body"}), 400
 
-    # Run the async function
-    solution = await generate_solution_python(user_query)
-    return jsonify({"solution": solution})
+        # Run the async function
+        solution = await generate_solution_python(user_query)
+        return jsonify({"solution": solution})
+
+    except Exception as e:
+        # Catch any unexpected errors during request processing or function call
+        print(f"Error in /generate endpoint: {e}")
+        return jsonify({"error": f"Internal server error: {e}"}), 500
 
 if __name__ == '__main__':
     # Hugging Face Spaces typically expect the app to run on port 7860