modified ui

interface.py

@@ -52,8 +52,8 @@ def check_api_health():
     except:
         return False
 
-def process_web_rag(url, prompt, data_source, progress=gr.Progress()):
-    """Main processing function with progress tracking"""
+def process_multiple_links(url, prompt):
+    """Process multiple links from a webpage"""
     if not url or not prompt:
         return "❌ Error: Please provide both URL and prompt", "", ""
     
@@ -62,28 +62,34 @@ def process_web_rag(url, prompt, data_source, progress=gr.Progress()):
         return "❌ Error: FastAPI service is not available. Please wait a moment and try again.", "", ""
     
     try:
-        progress(0.1, desc="Starting process...")
+        links = extract_links(url)
+        sample_links = links[:5]
+        file_path = extract_text(sample_links)
+        result = perform_rag(file_path, prompt)
         
-        if data_source == "Multiple links (first 5)":
-            progress(0.2, desc="🔍 Extracting links from webpage...")
-            links = extract_links(url)
-            sample_links = links[:5]
-            
-            progress(0.4, desc="📄 Extracting text from multiple pages...")
-            file_path = extract_text(sample_links)
-            
-            status_msg = f"✅ Processed {len(sample_links)} pages from {len(links)} total links found"
-        else:
-            progress(0.3, desc="📄 Extracting text from homepage...")
-            file_path = extract_text([url])
-            status_msg = "✅ Processed homepage content"
+        status_msg = f"✅ Processed {len(sample_links)} pages from {len(links)} total links found"
+        response_text = f"**Query:** {result['user_query']}\n\n**Response:** {result['assistant_response']}"
+        sources_text = result['sources']
         
-        progress(0.7, desc="🤖 Performing RAG analysis...")
-        result = perform_rag(file_path, prompt)
+        return status_msg, response_text, sources_text
         
-        progress(1.0, desc="✅ Complete!")
+    except Exception as e:
+        return f"❌ Error: {str(e)}", "", ""
+
+def process_homepage_only(url, prompt):
+    """Process homepage content only"""
+    if not url or not prompt:
+        return "❌ Error: Please provide both URL and prompt", "", ""
+    
+    # Check API health first
+    if not check_api_health():
+        return "❌ Error: FastAPI service is not available. Please wait a moment and try again.", "", ""
+    
+    try:
+        file_path = extract_text([url])
+        result = perform_rag(file_path, prompt)
         
-        # Format the response
+        status_msg = "✅ Processed homepage content"
         response_text = f"**Query:** {result['user_query']}\n\n**Response:** {result['assistant_response']}"
         sources_text = result['sources']
         
@@ -92,175 +98,121 @@ def process_web_rag(url, prompt, data_source, progress=gr.Progress()):
     except Exception as e:
         return f"❌ Error: {str(e)}", "", ""
 
-# Custom CSS for modern styling
-custom_css = """
-.gradio-container {
-    max-width: 900px !important;
-    margin: auto !important;
-}
-
-.header-text {
-    text-align: center;
-    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-    -webkit-background-clip: text;
-    -webkit-text-fill-color: transparent;
-    font-size: 2.5em;
-    font-weight: bold;
-    margin-bottom: 0.5em;
-}
-
-.description-text {
-    text-align: center;
-    color: #666;
-    font-size: 1.1em;
-    margin-bottom: 2em;
-}
-
-.input-group {
-    background: #f8f9fa;
-    padding: 1.5em;
-    border-radius: 12px;
-    margin: 1em 0;
-    border: 1px solid #e9ecef;
-}
-
-.output-group {
-    background: #ffffff;
-    border-radius: 12px;
-    border: 1px solid #dee2e6;
-    margin: 1em 0;
-}
-
-.status-box {
-    padding: 1em;
-    border-radius: 8px;
-    margin: 0.5em 0;
-}
-
-.status-success {
-    background-color: #d4edda;
-    border-color: #c3e6cb;
-    color: #155724;
-}
-
-.status-error {
-    background-color: #f8d7da;
-    border-color: #f5c6cb;
-    color: #721c24;
-}
-"""
+def extract_links_only(url):
+    """Extract and display links from a webpage"""
+    if not url:
+        return "❌ Error: Please provide a URL"
+    
+    # Check API health first
+    if not check_api_health():
+        return "❌ Error: FastAPI service is not available. Please wait a moment and try again."
+    
+    try:
+        links = extract_links(url)
+        links_text = "\n".join([f"• {link}" for link in links])
+        return f"✅ Found {len(links)} links:\n\n{links_text}"
+        
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
 
-# Create the Gradio interface
-with gr.Blocks(css=custom_css, title="Web RAG System", theme=gr.themes.Soft()) as app:
-    # Header
-    gr.HTML("""
-        <div class="header-text">🌐 Web RAG System</div>
-        <div class="description-text">
-            Extract content from web pages and ask questions using AI-powered retrieval
-        </div>
-    """)
+def extract_text_only(url):
+    """Extract and display text from a webpage"""
+    if not url:
+        return "❌ Error: Please provide a URL"
+    
+    # Check API health first
+    if not check_api_health():
+        return "❌ Error: FastAPI service is not available. Please wait a moment and try again."
     
+    try:
+        file_path = extract_text([url])
+        return f"✅ Text extracted and saved to: {file_path}"
+        
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
+
+# Create individual interfaces for each tab
+multiple_links_demo = gr.Interface(
+    process_multiple_links,
+    inputs=[
+        gr.Textbox(label="🔗 Website URL", placeholder="https://example.com"),
+        gr.Textbox(label="❓ Your Question", placeholder="What is this website about?", lines=3)
+    ],
+    outputs=[
+        gr.Textbox(label="📊 Status"),
+        gr.Textbox(label="🤖 AI Response", lines=8),
+        gr.Textbox(label="📚 Sources", lines=6)
+    ],
+    title=" \n🌐 Multiple Links Analysis",
+    allow_flagging="never"
+)
+
+homepage_only_demo = gr.Interface(
+    process_homepage_only,
+    inputs=[
+        gr.Textbox(label="🔗 Website URL", placeholder="https://example.com"),
+        gr.Textbox(label="❓ Your Question", placeholder="What is this website about?", lines=3)
+    ],
+    outputs=[
+        gr.Textbox(label="📊 Status"),
+        gr.Textbox(label="🤖 AI Response", lines=8),
+        gr.Textbox(label="📚 Sources", lines=6)
+    ],
+    title=" \n🏠 Homepage Only Analysis",
+    allow_flagging="never"
+)
+
+# Tab names
+tab_names = ["Multiple Links Analysis", "Homepage Only Analysis"]
+
+# Main interface with tabs
+with gr.Blocks(theme=gr.themes.Soft()) as interface:
     with gr.Row():
-        with gr.Column(scale=1):
-            # Input section
-            gr.HTML('<div style="font-size: 1.2em; font-weight: bold; margin-bottom: 1em;">📝 Input Configuration</div>')
+        with gr.Column(scale=1, min_width=1200):
+            gr.Markdown("## \n\n<style>h1, h2, h3, h4, h5, h6 { font-size: 40px !important; text-align:center !important; display: block !important;}</style>", label="")
+            gr.HTML("""
+                <div style="text-align: center; margin: 2em 0;">
+                    <h1 style="font-size: 3em; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); -webkit-background-clip: text; -webkit-text-fill-color: transparent; margin-bottom: 0.5em;">
+                        🌐 Web RAG System
+                    </h1>
+                    <p style="font-size: 1.2em; color: #666; margin-bottom: 1em;">
+                        Extract content from web pages and ask questions using AI-powered retrieval
+                    </p>
+                </div>
+            """)
             
-            url_input = gr.Textbox(
-                label="🔗 Website URL",
-                placeholder="https://example.com",
-                info="Enter the URL you want to analyze"
-            )
-            
-            prompt_input = gr.Textbox(
-                label="❓ Your Question",
-                placeholder="What is this website about?",
-                lines=3,
-                info="Ask any question about the content"
-            )
+            # Add custom CSS for increasing font size of text areas and toggles
+            gr.HTML("""
+                <style>
+                    textarea, input[type="text"], .gr-button {
+                        font-size: 1.2em !important;
+                    }
+                </style>
+            """)
             
-            data_source = gr.Radio(
-                choices=["Multiple links (first 5)", "Homepage only"],
-                value="Multiple links (first 5)",
-                label="📊 Data Source",
-                info="Choose how much content to analyze"
+            tabbed_interface = gr.TabbedInterface(
+                [multiple_links_demo, homepage_only_demo], 
+                tab_names
             )
             
-            process_btn = gr.Button(
-                "🚀 Analyze Website",
-                variant="primary",
-                size="lg"
-            )
-    
-    # Output section
-    gr.HTML('<div style="font-size: 1.2em; font-weight: bold; margin: 2em 0 1em 0;">📋 Results</div>')
-    
-    status_output = gr.Textbox(
-        label="📊 Processing Status",
-        interactive=False,
-        show_label=True
-    )
-    
-    with gr.Row():
-        with gr.Column(scale=2):
-            response_output = gr.Textbox(
-                label="🤖 AI Response",
-                lines=8,
-                interactive=False,
-                show_label=True
-            )
-        
-        with gr.Column(scale=1):
-            sources_output = gr.Textbox(
-                label="📚 Sources",
-                lines=8,
-                interactive=False,
-                show_label=True
-            )
-    
-    # Example section
-    gr.HTML("""
-        <div style="margin-top: 2em; padding: 1.5em; background: #f8f9fa; border-radius: 12px; border-left: 4px solid #667eea;">
-            <h3 style="margin-top: 0; color: #333;">💡 Example Usage</h3>
-            <p><strong>URL:</strong> https://openai.com</p>
-            <p><strong>Question:</strong> What are the main products and services offered?</p>
-            <p><strong>Data Source:</strong> Multiple links (first 5)</p>
-        </div>
-    """)
-    
-    # Add a note about the system status
-    gr.HTML("""
-        <div style="margin-top: 1em; padding: 1em; background: #e3f2fd; border-radius: 8px; border-left: 4px solid #2196f3;">
-            <p style="margin: 0; color: #0d47a1;">
-                ℹ️ <strong>Note:</strong> If you encounter connection errors, please wait a moment for the system to initialize and try again.
-            </p>
-        </div>
-    """)
-    
-    # Connect the function
-    process_btn.click(
-        fn=process_web_rag,
-        inputs=[url_input, prompt_input, data_source],
-        outputs=[status_output, response_output, sources_output],
-        show_progress=True
-    )
-    
-    # Add keyboard shortcut
-    url_input.submit(
-        fn=process_web_rag,
-        inputs=[url_input, prompt_input, data_source],
-        outputs=[status_output, response_output, sources_output],
-        show_progress=True
-    )
-    
-    prompt_input.submit(
-        fn=process_web_rag,
-        inputs=[url_input, prompt_input, data_source],
-        outputs=[status_output, response_output, sources_output],
-        show_progress=True
-    )
-
+            # Add example and notes at the bottom
+            gr.HTML("""
+                <div style="margin-top: 2em; padding: 1.5em; background: #f8f9fa; border-radius: 12px; border-left: 4px solid #667eea;">
+                    <h3 style="margin-top: 0; color: #333;">💡 Example Usage</h3>
+                    <p><strong>URL:</strong> https://openai.com</p>
+                    <p><strong>Question:</strong> What are the main products and services offered?</p>
+                </div>
+                
+                <div style="margin-top: 1em; padding: 1em; background: #e3f2fd; border-radius: 8px; border-left: 4px solid #2196f3;">
+                    <p style="margin: 0; color: #0d47a1;">
+                        ℹ️ <strong>Note:</strong> If you encounter connection errors, please wait a moment for the system to initialize and try again.
+                    </p>
+                </div>
+            """)
+
+# Launch the interface
 if __name__ == "__main__":
-    app.launch(
+    interface.launch(
         server_name="0.0.0.0",
         server_port=7860,
         share=False,