Update app.py

app.py

@@ -4,9 +4,34 @@ import os
 from groq import Groq
 from duckduckgo_search import DDGS
 
+# List of Groq models (defined globally to avoid NameError)
+groq_models = [
+    "qwen-2.5-32b",
+    "qwen-2.5-coder-32b",
+    "qwen-qwq-32b",
+    "deepseek-r1-distill-qwen-32b",
+    "deepseek-r1-distill-llama-70b",
+    "gemma2-9b-it",
+    "distil-whisper-large-v3-en",
+    "llama-3-1-8b-instant",
+    "llama-3-2-11b-vision-preview",
+    "llama-3.2-1b-preview",
+    "llama-3.2-3b-preview",
+    "llama-3.2-90b-vision-preview",
+    "llama-3.3-70b-specdec",
+    "llama-3.3-70b-versatile",
+    "llama-guard-3-8b",
+    "llama3-70b-8192",
+    "llama3-8b-8192",
+    "mistral-saba-24b",
+    "mixtral-8x7b-32768"
+]
+
 # Function to convert JSON to formatted text
 def json_to_text(filename):
     try:
+        if not filename:
+            return "Please select a JSON file."
         with open(filename, 'r') as json_file:
             data = json.load(json_file)
         return json.dumps(data, indent=2)
@@ -26,8 +51,8 @@ def search_duckduckgo(query):
 
 # Function to chat with JSON content using Groq API and DuckDuckGo fallback
 def chat_with_json(api_key, model_name, filename, question):
-    if not api_key or not filename or not question:
-        return "Please provide an API key, select a JSON file, and enter a 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."
 
     try:
         client = Groq(api_key=api_key)
@@ -36,11 +61,15 @@ def chat_with_json(api_key, model_name, filename, question):
 
         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'."
 
-        response = client.chat.completions.create(
+        completion = client.chat.completions.create(
             model=model_name,
-            messages=[{"role": "user", "content": prompt}]
+            messages=[{"role": "user", "content": prompt}],
+            temperature=1,
+            max_completion_tokens=1024,
+            top_p=1,
+            stream=False  # Set to False for simplicity; can be True for streaming
         )
-        answer = response.choices[0].message.content
+        answer = completion.choices[0].message.content
 
         if 'SEARCH_WEB' in answer:
             return search_duckduckgo(question)
@@ -51,30 +80,34 @@ def chat_with_json(api_key, model_name, filename, question):
     except Exception as e:
         return f"Error with Groq API: {str(e)}"
 
-# List available JSON files
+# Function to get list of JSON files dynamically
 def get_json_files():
-    return [f for f in os.listdir('.') if f.endswith('.json')] or ["No JSON files found"]
+    json_files = [f for f in os.listdir('.') if f.endswith('.json')]
+    return json_files if json_files else ["No JSON files found"]
 
-# Gradio Interface
+# Gradio Interface for JSON to Text
 iface = gr.Interface(
     fn=json_to_text,
     inputs=gr.Dropdown(choices=get_json_files(), label="Select JSON File"),
     outputs=gr.Markdown(),
-    title="JSON to Text"
+    title="JSON to Text",
+    description="Convert a JSON file to formatted text."
 )
 
+# Gradio Interface for Chat with JSON
 chat_iface = gr.Interface(
     fn=chat_with_json,
     inputs=[
-        gr.Textbox(label="Enter Groq API Key", type="password"),  # Mask API key
+        gr.Textbox(label="Enter Groq API Key", type="password"),
         gr.Dropdown(choices=groq_models, label="Select Model"),
         gr.Dropdown(choices=get_json_files(), label="Select JSON File"),
         gr.Textbox(label="Ask your question")
     ],
     outputs=gr.Markdown(),
     title="Chat with JSON and DuckDuckGo",
-    description="If the JSON does not have a relevant answer, it will use DuckDuckGo search as fallback."
+    description="Ask questions about a JSON file. If the answer isn’t in the JSON, it will search DuckDuckGo."
 )
 
+# Launch the tabbed interface
 if __name__ == "__main__":
     gr.TabbedInterface([iface, chat_iface], ["JSON to Text", "Chat with JSON"]).launch()