Update app.py

app.py

@@ -82,28 +82,10 @@ def analyze_construction_image(image):
             logger.warning("Received empty response from API")
             return [("Image analysis request", "Error: Received empty response from API")]
 
-        # Parse the result
-        analysis_text = "Safety and Hazard Analysis of the Construction Site Image:\n\n"
-        
-        sections = ["Hazard Identification", "Categorization", "Detailed Description", "Resolution Steps"]
-        for section in sections:
-            start = result.find(section)
-            if start != -1:
-                end = result.find(next((s for s in sections if s != section and result.find(s) > start), None))
-                content = result[start:end if end != -1 else None].strip()
-                analysis_text += f"{content}\n\n"
-
-        if analysis_text == "Safety and Hazard Analysis of the Construction Site Image:\n\n":
-            logger.warning("No sections found in the response")
-            return [("Image analysis request", "Error: Unable to parse the response. No sections found.")]
-
         logger.info("Analysis completed successfully")
         
-        # Split the analysis into chunks if it's too long
-        max_chunk_length = 1000  # Adjust this value as needed
-        chunks = [analysis_text[i:i+max_chunk_length] for i in range(0, len(analysis_text), max_chunk_length)]
-        
-        return [("Image analysis request", chunk) for chunk in chunks]
+        # Return the result directly, without any parsing
+        return [("Image analysis request", result)]
     except Exception as e:
         logger.error(f"Error during image analysis: {str(e)}")
         logger.error(traceback.format_exc())
@@ -161,7 +143,6 @@ custom_css = """
 .groq-badge { position: fixed; bottom: 10px; right: 10px; background-color: #f39c12; color: white; padding: 5px 10px; border-radius: 5px; font-weight: bold; }
 """
 
-# Create the Gradio interface
 # Create the Gradio interface
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as iface:
     gr.HTML(
@@ -190,17 +171,16 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as iface:
                 )
                 clear = gr.Button("🗑️ Clear Chat", elem_classes="clear-button")
 
-    def append_to_chat(history, new_messages):
-        if history is None:
-            history = []
-        history.extend(new_messages)
+    def update_chat(history, new_message):
+        history = history or []
+        history.append(new_message)
         return history
 
     analyze_button.click(
         analyze_construction_image,
         inputs=[image_input],
         outputs=[chatbot],
-        postprocess=append_to_chat
+        postprocess=lambda x: update_chat(chatbot.value, x[0])
     )
 
     msg.submit(chat_about_image, [msg, chatbot], [msg, chatbot])