Update app.py

app.py

@@ -5,7 +5,7 @@ from gradio_client import Client, handle_file
 API_URL = "ellamind/sui-demo"
 
 # ============================================================================
-# Theme & Styling (Fixed for Dark Mode)
+# Theme & Styling
 # ============================================================================
 
 THEME = gr.themes.Soft(
@@ -45,7 +45,6 @@ CSS = """
     padding-bottom: 0.5rem;
 }
 
-/* Thinking/Reasoning block inside the summary markdown */
 .thinking-section {
     background: var(--secondary-50);
     border-left: 4px solid var(--primary-500);
@@ -56,7 +55,6 @@ CSS = """
     color: var(--body-text-color-subdued);
 }
 
-/* Specific fix for dark mode reasoning block */
 [data-theme="dark"] .thinking-section {
     background: var(--neutral-800);
 }
@@ -86,17 +84,16 @@ CSS = """
 # ============================================================================
 
 def summarize_proxy(pdf_file, language, words, custom_instruction):
-    """Calls the backend Space API and yields results."""
+    """Calls the backend Space API and yields exactly 3 results."""
     if not pdf_file:
+        # Yielding 3 values to match the interface components
         yield "Please upload a PDF document.", "", gr.update(visible=False)
         return
 
     try:
-        # Initialize the client pointing to the demo space
         client = Client(API_URL)
         
-        # Use the 'summarize' API endpoint (index 0 in your config)
-        # Note: handle_file is necessary for Gradio 5.0+ to process local files correctly
+        # We start the stream. The backend yields 3 objects: (summary, sources, accordion_update)
         job = client.submit(
             pdf_file=handle_file(pdf_file),
             language=language,
@@ -105,20 +102,25 @@ def summarize_proxy(pdf_file, language, words, custom_instruction):
             api_name="/summarize"
         )
 
-        # Iterate through the generator responses for streaming
         for result in job:
-            # Result is a tuple: (formatted_output, sources_md, accordion_update)
-            yield result
+            # Result from the client should be a list/tuple of 3 items
+            # But let's be safe and ensure it matches the 3 outputs
+            if len(result) == 3:
+                yield result[0], result[1], result[2]
+            else:
+                # Fallback in case backend returns unexpected shape
+                yield result[0], result[1], gr.update(visible=True)
 
     except Exception as e:
-        yield f"### API Error\n{str(e)}", "Please ensure the backend Space is active.", gr.update(visible=False)
+        yield f"### API Error\n{str(e)}", "Ensure the backend Space is active.", gr.update(visible=False)
 
 # ============================================================================
 # Interface
 # ============================================================================
 
 def create_app():
-    with gr.Blocks(title="sui-1-24b", theme=THEME, css=CSS) as app:
+    # Removed theme and css from constructor for Gradio 6.0 compatibility
+    with gr.Blocks(title="sui-1-24b") as app:
         
         gr.HTML("""
         <div class="main-header">
@@ -128,7 +130,6 @@ def create_app():
         """)
 
         with gr.Row(equal_height=False):
-            # Left: Inputs
             with gr.Column(scale=2):
                 gr.Markdown("### Document", elem_classes=["section-title"])
                 pdf_input = gr.File(label="Upload PDF", file_types=[".pdf"], type="filepath")
@@ -148,14 +149,11 @@ def create_app():
                 
                 generate_btn = gr.Button("Generate Summary", variant="primary", elem_classes=["generate-btn"])
 
-            # Right: Outputs
             with gr.Column(scale=3):
                 gr.Markdown("### Summary", elem_classes=["section-title"])
-                summary_output = gr.Markdown(
-                    value="*Upload a PDF to begin.*",
-                    container=True
-                )
+                summary_output = gr.Markdown(value="*Upload a PDF to begin.*")
 
+                # Note: This is output #3
                 with gr.Accordion("View Source Citations", open=False, visible=False) as sources_accordion:
                     sources_output = gr.Markdown(elem_classes=["sources-panel"])
 
@@ -174,4 +172,5 @@ def create_app():
     return app
 
 if __name__ == "__main__":
-    create_app().launch()
+    # Theme and CSS passed here to comply with Gradio 6.0 warning
+    create_app().launch(theme=THEME, css=CSS)