Update app.py

app.py

@@ -116,7 +116,7 @@ def search_duckduckgo(query):
     except Exception as e:
         return f"Error during DuckDuckGo search: {str(e)}"
 
-# Function to chat with JSON content using Groq API and DuckDuckGo fallback
+# Function to chat with JSON content using Groq API with smarter handling
 def chat_with_json(api_key, model_name, filename, question):
     if not api_key or not model_name or not filename or not question:
         return "Please provide all required inputs: API key, model, JSON file, and question."
@@ -126,7 +126,16 @@ def chat_with_json(api_key, model_name, filename, question):
         with open(filename, 'r') as json_file:
             data = json.load(json_file)
 
-        prompt = f"JSON Data:\n{json.dumps(data, indent=2)}\n\nUser question: {question}\n\nAnswer based on the JSON data if possible. If the question cannot be answered based on the JSON, respond with 'SEARCH_WEB'."
+        # Enhanced prompt to guide the LLM
+        prompt = (
+            f"Here is the JSON data about a company:\n{json.dumps(data, indent=2)}\n\n"
+            f"User question: {question}\n\n"
+            "Answer the question based solely on the provided JSON data. "
+            "If the exact information is not available in the JSON, provide a thoughtful response using related information from the JSON. "
+            "For example, if asked about the CEO and the CEO is not listed, state that the CEO is not specified and provide other relevant details like the company's headquarters, founded year, or business description. "
+            "Do NOT suggest searching the web unless the user explicitly asks for a web search or the question is completely unrelated to the JSON content (e.g., asking about weather or unrelated topics). "
+            "If a web search is needed, respond with 'I can search the web for this information if you'd like. Please confirm.'"
+        )
 
         completion = client.chat.completions.create(
             model=model_name,
@@ -138,8 +147,9 @@ def chat_with_json(api_key, model_name, filename, question):
         )
         answer = completion.choices[0].message.content
 
-        if 'SEARCH_WEB' in answer:
-            return search_duckduckgo(question)
+        # Check if the LLM suggests a web search
+        if "I can search the web" in answer:
+            return answer  # Let the user decide if they want a web search
         return answer
 
     except (FileNotFoundError, json.JSONDecodeError) as e:
@@ -172,7 +182,7 @@ chat_iface = gr.Interface(
     ],
     outputs=gr.Markdown(),
     title="Chat with JSON and DuckDuckGo",
-    description="Ask questions about a JSON file. If the answer isn’t in the JSON, it will search DuckDuckGo."
+    description="Ask questions about a JSON file. The system will answer based on the JSON data, providing related insights if the exact answer is missing. Web searches are performed only if explicitly requested."
 )
 
 # Launch the tabbed interface