Update app.py

app.py

@@ -47,9 +47,11 @@ llm = LiteLLMModel(model_id="gemini/gemini-pro", api_key=gemini_api_key)
 def call_llm(prompt: str) -> str:
     """
     Helper to call the LLM with a prompt, handling response extraction.
+    Passes the prompt as a dictionary as expected by LiteLLMModel.
     """
     try:
-        result = llm(prompt)  # Directly call the model with the prompt
+        # Pass prompt in a dictionary format expected by the model
+        result = llm({"prompt": prompt})
         return result if isinstance(result, str) else str(result)
     except Exception as e:
         logger.error(f"LLM call error: {e}")