Update app.py

app.py

@@ -369,154 +369,148 @@ def process_full_pipeline(image):
     if image is None:
         return [], None
     
-    try:
-        # Get original image size for later restoration
-        original_img = Image.open(image)
-        original_size = original_img.size
-        
-        # Initialize chat history for UI
-        chat_history = [("Image uploaded for analysis", None)]
-        
-        # Step 1: Get streaming LLM response
-        streamer = get_llm_response_streaming(image)
-        
-        # Stream the response to UI with real-time updates
-        full_response = ""
-        in_reason = False
-        in_answer = False
-        reason_displayed = False
-        answer_displayed = False
-        reasoning_added = False  # Track if reasoning entry was added
+
+    # Get original image size for later restoration
+    original_img = Image.open(image)
+    original_size = original_img.size
+    
+    # Initialize chat history for UI
+    chat_history = [("Image uploaded for analysis", None)]
+    
+    # Step 1: Get streaming LLM response
+    streamer = get_llm_response_streaming(image)
+    
+    # Stream the response to UI with real-time updates
+    full_response = ""
+    in_reason = False
+    in_answer = False
+    reason_displayed = False
+    answer_displayed = False
+    reasoning_added = False  # Track if reasoning entry was added
+    
+    for new_text in streamer:
+        full_response += new_text
         
-        for new_text in streamer:
-            full_response += new_text
+        # Check if we're entering reason section or if we need to start showing content
+        if ('<reason>' in full_response and not in_reason and not reason_displayed) or (not reasoning_added and not in_reason and not reason_displayed):
+            in_reason = True
+            reasoning_added = True
+            
+            if '<reason>' in full_response:
+                # Extract content after <reason>
+                reason_start = full_response.find('<reason>') + len('<reason>')
+                reason_content = full_response[reason_start:].strip()
+            else:
+                # Show all content as reasoning if no tag yet
+                reason_content = full_response.strip()
             
-            # Check if we're entering reason section or if we need to start showing content
-            if ('<reason>' in full_response and not in_reason and not reason_displayed) or (not reasoning_added and not in_reason and not reason_displayed):
-                in_reason = True
-                reasoning_added = True
+            # Add reasoning to chat history
+            chat_history.append((None, f"**🤔 Analysis & Reasoning:**\n\n{reason_content}"))
+            yield chat_history, None
+        
+        # If we're in reason section, update content
+        elif in_reason and not reason_displayed:
+            # Check if reason section is complete
+            if '</reason>' in full_response:
+                # Extract complete reason content
+                reason_start = full_response.find('<reason>') + len('<reason>')
+                reason_end = full_response.find('</reason>')
+                reason_content = full_response[reason_start:reason_end].strip()
                 
+                # Update chat history with complete reason
+                chat_history[1] = (None, f"**🤔 Analysis & Reasoning:**\n\n{reason_content}")
+                reason_displayed = True
+                in_reason = False
+                yield chat_history, None
+            else:
+                # Continue streaming reason content
                 if '<reason>' in full_response:
-                    # Extract content after <reason>
                     reason_start = full_response.find('<reason>') + len('<reason>')
                     reason_content = full_response[reason_start:].strip()
                 else:
-                    # Show all content as reasoning if no tag yet
                     reason_content = full_response.strip()
                 
-                # Add reasoning to chat history
-                chat_history.append((None, f"**🤔 Analysis & Reasoning:**\n\n{reason_content}"))
+                # Update chat history with partial reason
+                chat_history[1] = (None, f"**🤔 Analysis & Reasoning:**\n\n{reason_content}")
                 yield chat_history, None
+        
+        # Check if we're entering answer section
+        elif '<answer>' in full_response and not in_answer and not answer_displayed and reason_displayed:
+            in_answer = True
+            # Extract content after <answer>
+            answer_start = full_response.find('<answer>') + len('<answer>')
+            answer_content = full_response[answer_start:]
             
-            # If we're in reason section, update content
-            elif in_reason and not reason_displayed:
-                # Check if reason section is complete
-                if '</reason>' in full_response:
-                    # Extract complete reason content
-                    reason_start = full_response.find('<reason>') + len('<reason>')
-                    reason_end = full_response.find('</reason>')
-                    reason_content = full_response[reason_start:reason_end].strip()
-                    
-                    # Update chat history with complete reason
-                    chat_history[1] = (None, f"**🤔 Analysis & Reasoning:**\n\n{reason_content}")
-                    reason_displayed = True
-                    in_reason = False
-                    yield chat_history, None
-                else:
-                    # Continue streaming reason content
-                    if '<reason>' in full_response:
-                        reason_start = full_response.find('<reason>') + len('<reason>')
-                        reason_content = full_response[reason_start:].strip()
-                    else:
-                        reason_content = full_response.strip()
-                    
-                    # Update chat history with partial reason
-                    chat_history[1] = (None, f"**🤔 Analysis & Reasoning:**\n\n{reason_content}")
-                    yield chat_history, None
-            
-            # Check if we're entering answer section
-            elif '<answer>' in full_response and not in_answer and not answer_displayed and reason_displayed:
-                in_answer = True
-                # Extract content after <answer>
+            # Add partial answer to chat history
+            models = extract_models_from_answer(answer_content)
+            beautified = beautify_recommended_actions(answer_content, models)
+            chat_history.append((None, beautified))
+            yield chat_history, None
+        
+        # If we're in answer section, update content
+        elif in_answer and not answer_displayed:
+            # Check if answer section is complete
+            if '</answer>' in full_response:
+                # Extract complete answer content
                 answer_start = full_response.find('<answer>') + len('<answer>')
-                answer_content = full_response[answer_start:]
+                answer_end = full_response.find('</answer>')
+                answer_content = full_response[answer_start:answer_end].strip()
                 
-                # Add partial answer to chat history
+                # Parse and process final answer
                 models = extract_models_from_answer(answer_content)
                 beautified = beautify_recommended_actions(answer_content, models)
-                chat_history.append((None, beautified))
+                chat_history[2] = (None, beautified)
+                answer_displayed = True
+                in_answer = False
                 yield chat_history, None
-            
-            # If we're in answer section, update content
-            elif in_answer and not answer_displayed:
-                # Check if answer section is complete
-                if '</answer>' in full_response:
-                    # Extract complete answer content
-                    answer_start = full_response.find('<answer>') + len('<answer>')
-                    answer_end = full_response.find('</answer>')
-                    answer_content = full_response[answer_start:answer_end].strip()
-                    
-                    # Parse and process final answer
-                    models = extract_models_from_answer(answer_content)
-                    beautified = beautify_recommended_actions(answer_content, models)
-                    chat_history[2] = (None, beautified)
-                    answer_displayed = True
-                    in_answer = False
+                
+                # Process image with tools
+                if models:
+                    chat_history.append((None, "**🔄 Processing image...**"))
                     yield chat_history, None
                     
-                    # Process image with tools
-                    if models:
-                        chat_history.append((None, "**🔄 Processing image...**"))
-                        yield chat_history, None
-                        
-                        processed_image = process_image_with_tools(image, models, original_size)
-                        chat_history[-1] = (None, "**✅ Processing Complete!**")
-                        yield chat_history, processed_image
-                        return
-                    else:
-                        chat_history.append((None, "**❌ No valid models found in the response**"))
-                        yield chat_history, None
-                        return
+                    processed_image = process_image_with_tools(image, models, original_size)
+                    chat_history[-1] = (None, "**✅ Processing Complete!**")
+                    yield chat_history, processed_image
+                    return
                 else:
-                    # Continue streaming answer content
-                    answer_start = full_response.find('<answer>') + len('<answer>')
-                    answer_content = full_response[answer_start:].strip()
-                    
-                    # Update chat history with partial answer
-                    models = extract_models_from_answer(answer_content)
-                    beautified = beautify_recommended_actions(answer_content, models)
-                    chat_history[2] = (None, beautified)
+                    chat_history.append((None, "**❌ No valid models found in the response**"))
                     yield chat_history, None
-        
-        # Fallback if streaming completes without proper tags
-        if not answer_displayed:
-            reason, answer = parse_llm_response(full_response)
-            models = extract_models_from_answer(answer)
-            
-            chat_history = [
-                ("Image uploaded for analysis", None),
-                (None, f"**🤔 Analysis & Reasoning:**\n\n{reason}"),
-                (None, beautify_recommended_actions(answer, models))
-            ]
-            
-            if models:
-                chat_history.append((None, "**🔄 Processing image...**"))
-                yield chat_history, None
-                
-                processed_image = process_image_with_tools(image, models, original_size)
-                chat_history[-1] = (None, "**✅ Processing Complete!**")
-                yield chat_history, processed_image
+                    return
             else:
-                chat_history.append((None, "**❌ No valid models found in the response**"))
+                # Continue streaming answer content
+                answer_start = full_response.find('<answer>') + len('<answer>')
+                answer_content = full_response[answer_start:].strip()
+                
+                # Update chat history with partial answer
+                models = extract_models_from_answer(answer_content)
+                beautified = beautify_recommended_actions(answer_content, models)
+                chat_history[2] = (None, beautified)
                 yield chat_history, None
-            
-    except Exception as e:
-        error_msg = f"Error: {str(e)}"
+    
+    # Fallback if streaming completes without proper tags
+    if not answer_displayed:
+        reason, answer = parse_llm_response(full_response)
+        models = extract_models_from_answer(answer)
+        
         chat_history = [
             ("Image uploaded for analysis", None),
-            (None, f"**❌ Error occurred:**\n\n{error_msg}")
+            (None, f"**🤔 Analysis & Reasoning:**\n\n{reason}"),
+            (None, beautify_recommended_actions(answer, models))
         ]
-        yield chat_history, None
+        
+        if models:
+            chat_history.append((None, "**🔄 Processing image...**"))
+            yield chat_history, None
+            
+            processed_image = process_image_with_tools(image, models, original_size)
+            chat_history[-1] = (None, "**✅ Processing Complete!**")
+            yield chat_history, processed_image
+        else:
+            chat_history.append((None, "**❌ No valid models found in the response**"))
+            yield chat_history, None
+            
+    
 
 # Create Gradio interface
 def create_interface():