Update app.py

app.py

@@ -53,7 +53,7 @@ RESPONSE FORMAT:
 
 Remember: Your goal is to be the most reliable, up-to-date source of information possible."""
 
-    def query_compound_model(self, query, model, temperature=0.7):
+    def query_compound_model(self, query, model, temperature=0.7, custom_system_prompt=None):
         """Query the compound model and return response with tool execution info"""
         if not self.client:
             return "❌ Please set a valid API key first.", None, None
@@ -61,11 +61,14 @@ Remember: Your goal is to be the most reliable, up-to-date source of information
         try:
             start_time = time.time()
             
+            # Use custom system prompt if provided
+            system_prompt = custom_system_prompt if custom_system_prompt else self.get_system_prompt()
+            
             chat_completion = self.client.chat.completions.create(
                 messages=[
                     {
                         "role": "system",
-                        "content": self.get_system_prompt()
+                        "content": system_prompt
                     },
                     {
                         "role": "user",
@@ -74,7 +77,7 @@ Remember: Your goal is to be the most reliable, up-to-date source of information
                 ],
                 model=model,
                 temperature=temperature,
-                max_tokens=1000
+                max_tokens=1500
             )
             
             end_time = time.time()
@@ -83,7 +86,7 @@ Remember: Your goal is to be the most reliable, up-to-date source of information
             # Extract response
             response_content = chat_completion.choices[0].message.content
             
-            # Check for executed tools
+            # Check for executed tools - Fixed the error here
             executed_tools = getattr(chat_completion.choices[0].message, 'executed_tools', None)
             
             # Format tool execution info
@@ -95,21 +98,36 @@ Remember: Your goal is to be the most reliable, up-to-date source of information
             return f"❌ Error querying model: {str(e)}", None, None
     
     def format_tool_info(self, executed_tools):
-        """Format executed tools information for display"""
+        """Format executed tools information for display - FIXED"""
         if not executed_tools:
             return "🔍 **Tools Used:** None (Used existing knowledge)"
         
         tool_info = "🔍 **Tools Used:**\n"
         for i, tool in enumerate(executed_tools, 1):
-            tool_name = tool.get('name', 'Unknown')
-            tool_info += f"{i}. **{tool_name}**\n"
-            
-            # Add tool parameters if available
-            if 'parameters' in tool:
-                params = tool['parameters']
-                if isinstance(params, dict):
-                    for key, value in params.items():
-                        tool_info += f"   - {key}: {value}\n"
+            try:
+                # Handle different tool object types
+                if hasattr(tool, 'name'):
+                    tool_name = tool.name
+                elif hasattr(tool, 'tool_name'):
+                    tool_name = tool.tool_name
+                elif isinstance(tool, dict):
+                    tool_name = tool.get('name', 'Unknown')
+                else:
+                    tool_name = str(tool)
+                
+                tool_info += f"{i}. **{tool_name}**\n"
+                
+                # Add tool parameters if available
+                if hasattr(tool, 'parameters'):
+                    params = tool.parameters
+                    if isinstance(params, dict):
+                        for key, value in params.items():
+                            tool_info += f"   - {key}: {value}\n"
+                elif hasattr(tool, 'input'):
+                    tool_info += f"   - Input: {tool.input}\n"
+                    
+            except Exception as e:
+                tool_info += f"{i}. **Tool {i}** (Error parsing details)\n"
         
         return tool_info
     
@@ -159,10 +177,193 @@ Remember: Your goal is to be the most reliable, up-to-date source of information
                 "Verify recent claims about electric vehicle sales"
             ]
         }
+    
+    def get_custom_prompt_examples(self):
+        """Return custom system prompt examples"""
+        return {
+            "🎯 Fact-Checker": "You are a fact-checker. Always verify claims with multiple sources and clearly indicate confidence levels in your assessments. Use phrases like 'highly confident', 'moderately confident', or 'requires verification' when presenting information.",
+            
+            "📊 News Analyst": "You are a news analyst. Focus on providing balanced, unbiased reporting with multiple perspectives on current events. Always present different viewpoints and avoid partisan language.",
+            
+            "💼 Financial Advisor": "You are a financial advisor. Provide accurate market data with context about trends and implications for investors. Always include disclaimers about market risks and the importance of professional financial advice.",
+            
+            "🔬 Research Assistant": "You are a research assistant specializing in scientific and technical information. Provide detailed, evidence-based responses with proper context about methodology and limitations of studies.",
+            
+            "🌍 Global News Correspondent": "You are a global news correspondent. Focus on international events and their interconnections. Provide cultural context and explain how events in one region might affect others.",
+            
+            "📈 Market Analyst": "You are a market analyst. Provide detailed financial analysis including technical indicators, market sentiment, and economic factors affecting price movements."
+        }
 
 def create_interface():
     fact_checker = RealTimeFactChecker()
     
+    # Custom CSS for beautiful styling
+    custom_css = """
+    <style>
+    .gradio-container {
+        max-width: 1400px !important;
+        margin: 0 auto;
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        min-height: 100vh;
+        font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+    }
+    
+    .main-header {
+        background: linear-gradient(135deg, #1e3c72 0%, #2a5298 100%);
+        color: white;
+        padding: 30px;
+        border-radius: 20px;
+        margin-bottom: 30px;
+        text-align: center;
+        box-shadow: 0 10px 30px rgba(0,0,0,0.3);
+    }
+    
+    .main-header h1 {
+        font-size: 2.5rem;
+        margin: 0;
+        text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
+    }
+    
+    .main-header p {
+        font-size: 1.2rem;
+        margin: 10px 0 0 0;
+        opacity: 0.9;
+    }
+    
+    .feature-card {
+        background: white;
+        border-radius: 15px;
+        padding: 25px;
+        margin: 20px 0;
+        box-shadow: 0 8px 25px rgba(0,0,0,0.1);
+        border: 1px solid #e1e8ed;
+        transition: transform 0.3s ease, box-shadow 0.3s ease;
+    }
+    
+    .feature-card:hover {
+        transform: translateY(-5px);
+        box-shadow: 0 15px 40px rgba(0,0,0,0.2);
+    }
+    
+    .example-grid {
+        display: grid;
+        grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
+        gap: 15px;
+        margin-top: 20px;
+    }
+    
+    .example-category {
+        background: #f8f9fa;
+        border-radius: 10px;
+        padding: 15px;
+        border-left: 4px solid #667eea;
+    }
+    
+    .example-category h4 {
+        margin: 0 0 10px 0;
+        color: #2d3748;
+        font-weight: 600;
+    }
+    
+    .status-success {
+        background: linear-gradient(135deg, #48bb78 0%, #38a169 100%);
+        color: white;
+        padding: 10px 15px;
+        border-radius: 8px;
+        font-weight: 500;
+    }
+    
+    .status-warning {
+        background: linear-gradient(135deg, #ed8936 0%, #dd6b20 100%);
+        color: white;
+        padding: 10px 15px;
+        border-radius: 8px;
+        font-weight: 500;
+    }
+    
+    .status-error {
+        background: linear-gradient(135deg, #f56565 0%, #e53e3e 100%);
+        color: white;
+        padding: 10px 15px;
+        border-radius: 8px;
+        font-weight: 500;
+    }
+    
+    .results-section {
+        background: white;
+        border-radius: 15px;
+        padding: 30px;
+        margin: 30px 0;
+        box-shadow: 0 8px 25px rgba(0,0,0,0.1);
+    }
+    
+    .tool-info {
+        background: #f7fafc;
+        border-left: 4px solid #4299e1;
+        padding: 15px;
+        border-radius: 8px;
+        margin: 15px 0;
+    }
+    
+    .performance-badge {
+        background: linear-gradient(135deg, #38b2ac 0%, #319795 100%);
+        color: white;
+        padding: 8px 15px;
+        border-radius: 20px;
+        font-weight: 500;
+        display: inline-block;
+        margin: 10px 0;
+    }
+    
+    .footer-section {
+        background: #2d3748;
+        color: white;
+        padding: 30px;
+        border-radius: 15px;
+        margin-top: 30px;
+        text-align: center;
+    }
+    
+    .footer-section a {
+        color: #63b3ed;
+        text-decoration: none;
+        font-weight: 500;
+    }
+    
+    .footer-section a:hover {
+        color: #90cdf4;
+        text-decoration: underline;
+    }
+    
+    .prompt-example {
+        background: #ebf8ff;
+        border: 1px solid #bee3f8;
+        border-radius: 8px;
+        padding: 12px;
+        margin: 8px 0;
+        cursor: pointer;
+        transition: all 0.3s ease;
+    }
+    
+    .prompt-example:hover {
+        background: #bee3f8;
+        transform: translateX(5px);
+    }
+    
+    .prompt-example-title {
+        font-weight: 600;
+        color: #2b6cb0;
+        margin-bottom: 5px;
+    }
+    
+    .prompt-example-text {
+        font-size: 0.9rem;
+        color: #4a5568;
+        line-height: 1.4;
+    }
+    </style>
+    """
+    
     def validate_api_key(api_key):
         if not api_key or api_key.strip() == "":
             return "❌ Please enter a valid API key", False
@@ -183,19 +384,10 @@ def create_interface():
             if not success:
                 return message, "", ""
         
-        # Use custom system prompt if provided
-        if system_prompt and system_prompt.strip():
-            original_prompt = fact_checker.get_system_prompt
-            fact_checker.get_system_prompt = lambda: system_prompt.strip()
-        
         response, tool_info, response_time = fact_checker.query_compound_model(
-            query.strip(), model, temperature
+            query.strip(), model, temperature, system_prompt.strip() if system_prompt else None
         )
         
-        # Restore original system prompt function
-        if system_prompt and system_prompt.strip():
-            fact_checker.get_system_prompt = original_prompt
-        
         # Format response with timestamp
         timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         formatted_response = f"**Query:** {query}\n\n**Response:**\n{response}\n\n---\n*Generated at {timestamp} in {response_time}s*"
@@ -208,21 +400,25 @@ def create_interface():
     def load_example(example_text):
         return example_text
     
+    def load_custom_prompt(prompt_text):
+        return prompt_text
+    
     # Create the Gradio interface
-    with gr.Blocks(title="Real-time Fact Checker & News Agent", theme=gr.themes.Ocean()) as demo:
-        gr.Markdown("""
-        # 🔍 Real-time Fact Checker & News Agent
-        
-        **Powered by Groq's Compound Models with Built-in Web Search**
+    with gr.Blocks(title="Real-time Fact Checker & News Agent", css=custom_css) as demo:
         
-        This application provides real-time information by automatically searching the web when needed. 
-        Enter your query below and get up-to-the-minute facts, news, and data!
+        # Header
+        gr.HTML("""
+        <div class="main-header">
+            <h1>🔍 Real-time Fact Checker & News Agent</h1>
+            <p>Powered by Groq's Compound Models with Built-in Web Search</p>
+        </div>
         """)
         
         with gr.Row():
             with gr.Column(scale=2):
                 # API Key section
                 with gr.Group():
+                    gr.HTML('<div class="feature-card">')
                     gr.Markdown("### 🔑 API Configuration")
                     api_key_input = gr.Textbox(
                         label="Groq API Key",
@@ -236,26 +432,57 @@ def create_interface():
                         interactive=False
                     )
                     validate_btn = gr.Button("Validate API Key", variant="secondary")
+                    gr.HTML('</div>')
                 
                 # Advanced options
                 with gr.Group():
+                    gr.HTML('<div class="feature-card">')
                     gr.Markdown("### ⚙️ Advanced Options")
-                    with gr.Accordion("System Prompt (Click to customize)", open=False):
+                    
+                    # Custom System Prompt Examples
+                    with gr.Accordion("📝 System Prompt Examples (Click to view)", open=False):
+                        gr.Markdown("**Click any example to load it as your system prompt:**")
+                        
+                        custom_prompts = fact_checker.get_custom_prompt_examples()
+                        for title, prompt in custom_prompts.items():
+                            with gr.Row():
+                                gr.HTML(f"""
+                                <div class="prompt-example" onclick="document.getElementById('system_prompt_input').value = '{prompt}'">
+                                    <div class="prompt-example-title">{title}</div>
+                                    <div class="prompt-example-text">{prompt[:100]}...</div>
+                                </div>
+                                """)
+                    
+                    with gr.Accordion("🔧 System Prompt (Click to customize)", open=False):
                         system_prompt_input = gr.Textbox(
                             label="System Prompt",
                             value=fact_checker.get_system_prompt(),
                             lines=8,
-                            info="Customize how the AI behaves and responds"
+                            info="Customize how the AI behaves and responds",
+                            elem_id="system_prompt_input"
                         )
                         reset_prompt_btn = gr.Button("Reset to Default", variant="secondary", size="sm")
+                        
+                        # Add buttons for each custom prompt
+                        gr.Markdown("**Quick Load Custom Prompts:**")
+                        custom_prompts = fact_checker.get_custom_prompt_examples()
+                        for title, prompt in custom_prompts.items():
+                            prompt_btn = gr.Button(title, variant="secondary", size="sm")
+                            prompt_btn.click(
+                                fn=lambda p=prompt: p,
+                                outputs=[system_prompt_input]
+                            )
+                    
+                    gr.HTML('</div>')
 
                 # Query section
                 with gr.Group():
+                    gr.HTML('<div class="feature-card">')
                     gr.Markdown("### 💭 Your Query")
                     query_input = gr.Textbox(
                         label="Ask anything that requires real-time information",
                         placeholder="e.g., What are the latest AI developments today?",
-                        lines=3
+                        lines=4
                     )
                     
                     with gr.Row():
@@ -276,25 +503,69 @@ def create_interface():
                     
                     submit_btn = gr.Button("🔍 Get Real-time Information", variant="primary", size="lg")
                     clear_btn = gr.Button("Clear", variant="secondary")
+                    gr.HTML('</div>')
             
             with gr.Column(scale=1):
-                # Example queries
+                # Example queries with tabs
                 with gr.Group():
+                    gr.HTML('<div class="feature-card">')
                     gr.Markdown("### 📝 Example Queries")
                     gr.Markdown("Click any example to load it:")
                     
                     examples = fact_checker.get_example_queries()
-                    for category, queries in examples.items():
-                        gr.Markdown(f"**{category}**")
-                        for query in queries:
+                    
+                    with gr.Accordion("📰 Latest News", open=True):
+                        for query in examples["📰 Latest News"]:
+                            example_btn = gr.Button(query, variant="secondary", size="sm")
+                            example_btn.click(
+                                fn=lambda q=query: q,
+                                outputs=[query_input]
+                            )
+                    
+                    with gr.Accordion("💰 Financial Data", open=False):
+                        for query in examples["💰 Financial Data"]:
+                            example_btn = gr.Button(query, variant="secondary", size="sm")
+                            example_btn.click(
+                                fn=lambda q=query: q,
+                                outputs=[query_input]
+                            )
+                    
+                    with gr.Accordion("🌤️ Weather Updates", open=False):
+                        for query in examples["🌤️ Weather Updates"]:
                             example_btn = gr.Button(query, variant="secondary", size="sm")
                             example_btn.click(
-                                fn=load_example,
-                                inputs=[gr.State(query)],
+                                fn=lambda q=query: q,
                                 outputs=[query_input]
                             )
+                    
+                    with gr.Accordion("🔬 Science & Technology", open=False):
+                        for query in examples["🔬 Science & Technology"]:
+                            example_btn = gr.Button(query, variant="secondary", size="sm")
+                            example_btn.click(
+                                fn=lambda q=query: q,
+                                outputs=[query_input]
+                            )
+                    
+                    with gr.Accordion("🏆 Sports & Entertainment", open=False):
+                        for query in examples["🏆 Sports & Entertainment"]:
+                            example_btn = gr.Button(query, variant="secondary", size="sm")
+                            example_btn.click(
+                                fn=lambda q=query: q,
+                                outputs=[query_input]
+                            )
+                    
+                    with gr.Accordion("🔍 Fact Checking", open=False):
+                        for query in examples["🔍 Fact Checking"]:
+                            example_btn = gr.Button(query, variant="secondary", size="sm")
+                            example_btn.click(
+                                fn=lambda q=query: q,
+                                outputs=[query_input]
+                            )
+                    
+                    gr.HTML('</div>')
         
         # Results section
+        gr.HTML('<div class="results-section">')
         gr.Markdown("### 📊 Results")
         
         with gr.Row():
@@ -315,6 +586,7 @@ def create_interface():
                     value="",
                     interactive=False
                 )
+        gr.HTML('</div>')
         
         # Event handlers
         validate_btn.click(
@@ -340,17 +612,24 @@ def create_interface():
         )
         
         # Footer
-        gr.Markdown("""
-        ---
-        ### 🔗 Useful Links
-        - [Groq Console](https://console.groq.com/) - Get your free API key
-        - [Groq Documentation](https://console.groq.com/docs/quickstart) - Learn more about Groq models
-        - [Compound Models Info](https://console.groq.com/docs/models) - Details about compound models
-        
-        ### 💡 Tips
-        - The compound models automatically use web search when real-time information is needed
-        - Try different temperature settings: 0.1 for factual queries, 0.7-0.9 for creative questions
-        - compound-beta is more capable but slower, compound-beta-mini is faster but less capable
+        gr.HTML("""
+        <div class="footer-section">
+            <h3>🔗 Useful Links</h3>
+            <p>
+                <a href="https://console.groq.com/" target="_blank">Groq Console</a> - Get your free API key<br>
+                <a href="https://console.groq.com/docs/quickstart" target="_blank">Groq Documentation</a> - Learn more about Groq models<br>
+                <a href="https://console.groq.com/docs/models" target="_blank">Compound Models Info</a> - Details about compound models
+            </p>
+            
+            <h3>💡 Tips</h3>
+            <ul style="text-align: left; display: inline-block;">
+                <li>The compound models automatically use web search when real-time information is needed</li>
+                <li>Try different temperature settings: 0.1 for factual queries, 0.7-0.9 for creative questions</li>
+                <li>compound-beta is more capable but slower, compound-beta-mini is faster but less capable</li>
+                <li>Use custom system prompts to specialize the AI for different types of queries</li>
+                <li>Check the Tool Execution Info to see when web search was used</li>
+            </ul>
+        </div>
         """)
     
     return demo