Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import sys
 import logging
+import re
 
 # Logging configuration
 logging.basicConfig(level=logging.INFO, 
@@ -455,7 +456,6 @@ def optimize_user_prompt_with_zephyr(user_idea, scene_info=None):
         logger.info("Preparing prompt for optimization...")
         
         # Detect language and adjust system prompt accordingly
-        import re
         non_english_pattern = re.compile(r'[^\x00-\x7F]+')
         has_non_english = bool(non_english_pattern.search(user_idea))
         
@@ -557,7 +557,6 @@ def refine_prompt_with_zephyr(current_prompt, feedback, chat_history, scene_info
 
     try:
         # Detect language of current prompt and feedback
-        import re
         non_english_pattern = re.compile(r'[^\x00-\x7F]+')
         has_non_english_prompt = bool(non_english_pattern.search(current_prompt))
         has_non_english_feedback = bool(non_english_pattern.search(feedback))
@@ -727,6 +726,72 @@ def get_debug_info():
         logger.error(f"Error generating debug info: {str(e)}")
         return f"Error: {str(e)}"
 
+# Nueva función para el chat conversacional
+def chat_with_ai(user_input, chat_history, last_prompt, scene_info=None):
+    """Process user message for chat-based prompt creation or refinement"""
+    logger.info(f"Processing chat message: {user_input}")
+    
+    if not user_input.strip():
+        return "", chat_history, last_prompt
+    
+    try:
+        # Check if this is refinement of an existing prompt or a new prompt
+        is_refinement = False
+        
+        if last_prompt:
+            # Check if input seems like feedback rather than a new prompt idea
+            # Common words used in feedback
+            feedback_keywords = ['more', 'less', 'add', 'change', 'make', 'remove', 
+                                'modify', 'update', 'adjust',
+                                # Spanish equivalents
+                                'más', 'menos', 'añadir', 'cambiar', 'hacer', 'quitar',
+                                'modificar', 'actualizar', 'ajustar']
+            
+            # Check for feedback keywords and short inputs which are likely feedback
+            first_word = user_input.strip().split(' ')[0].lower() if user_input.strip() else ""
+            if (first_word in feedback_keywords or 
+                len(user_input.split()) <= 4 or 
+                not user_input[0].isupper()):
+                is_refinement = True
+        
+        # Detect language
+        non_english_pattern = re.compile(r'[^\x00-\x7F]+')
+        has_non_english = bool(non_english_pattern.search(user_input))
+        
+        if is_refinement and last_prompt:
+            # Process as refinement to previous prompt
+            logger.info("Processing as refinement feedback")
+            refined_prompt, _ = refine_prompt_with_zephyr(last_prompt, user_input, [], scene_info)
+            
+            # Prepare response based on language
+            if has_non_english:
+                response_text = f"Aquí está tu prompt refinado:\n\n**{refined_prompt}**\n\n¿Quieres hacer más cambios?"
+            else:
+                response_text = f"Here's your refined prompt:\n\n**{refined_prompt}**\n\nAny other changes you'd like to make?"
+                
+            chat_history.append([user_input, response_text])
+            return "", chat_history, refined_prompt
+            
+        else:
+            # Process as new prompt creation
+            logger.info("Processing as new prompt idea")
+            optimized, _ = optimize_user_prompt_with_zephyr(user_input, scene_info)
+            
+            # Prepare response based on language
+            if has_non_english:
+                response_text = f"Aquí está tu prompt optimizado:\n\n**{optimized}**\n\nPuedes pedir cambios específicos o mejoras."
+            else:
+                response_text = f"Here's your optimized video prompt:\n\n**{optimized}**\n\nYou can ask for specific changes or improvements."
+                
+            chat_history.append([user_input, response_text])
+            return "", chat_history, optimized
+            
+    except Exception as e:
+        logger.error(f"Error in chat processing: {str(e)}")
+        error_message = f"Error processing your message: {str(e)}. Please try again with a simpler request."
+        chat_history.append([user_input, error_message])
+        return "", chat_history, last_prompt
+
 # Create the Gradio interface
 def create_interface():
     """Create the Gradio interface"""
@@ -748,6 +813,7 @@ def create_interface():
         # State variables
         scene_state = gr.State({})
         chat_history_state = gr.State([])
+        last_prompt_state = gr.State("")
 
         with gr.Tabs():
             # Tab 1: Learning Guide
@@ -793,57 +859,46 @@ def create_interface():
                         for i in range(3)
                     ]
 
-            # Tab 3: AI Prompt Generator
+            # Tab 3: AI Prompt Generator (Conversational)
             with gr.Tab("🤖 AI Prompt Generator"):
                 with gr.Row():
-                    with gr.Column():
-                        user_idea = gr.Textbox(
-                            label="Your Video Idea (any language)",
-                            placeholder="e.g., 'el personaje camina lentamente' or 'character walks slowly'",
-                            lines=3
-                        )
-                        optimize_btn = gr.Button("🚀 Generate Optimized Prompt", variant="primary")
-                        with gr.Row():
-                            retry_btn = gr.Button("🔄 Manual Generation Fallback", variant="secondary")
-                            model_status = gr.Textbox(
-                                label="Model Status",
-                                value="",
-                                interactive=False
-                            )
-                        optimized_prompt = gr.Textbox(
-                            label="AI-Optimized Video Prompt",
-                            lines=4,
-                            interactive=True,
-                            show_copy_button=True
-                        )
-                        # Basic test button
-                        test_btn = gr.Button("🔬 Test Basic Generation", variant="secondary")
-                        test_output = gr.Textbox(
-                            label="Basic Generation Test",
-                            lines=2,
-                            interactive=False
-                        )
-                    with gr.Column():
-                        gr.Markdown("### 🔄 Refine Your Prompt")
-                        feedback_input = gr.Textbox(
-                            label="Feedback/Changes",
-                            placeholder="e.g., 'make it more dramatic' or 'add camera movement'",
-                            lines=2
-                        )
-                        refine_btn = gr.Button("🔄 Refine Prompt")
-                        # Chat history
-                        with gr.Accordion("💬 Refinement History", open=False):
-                            chat_display = gr.Chatbot(height=300, type='messages')
-                        
-                        # Model status and debug info
-                        with gr.Accordion("🔧 Debug Info", open=False):
-                            debug_info = gr.Textbox(
-                                label="Debug Information",
-                                value="Click 'Get Debug Info' to see model status",
-                                lines=8,
-                                interactive=False
-                            )
-                            debug_btn = gr.Button("Get Debug Info")
+                    # Conversation history
+                    chat_display = gr.Chatbot(
+                        height=450,
+                        show_copy_button=True,
+                        label="AI Conversation"
+                    )
+                
+                with gr.Row():
+                    # Message input
+                    user_message = gr.Textbox(
+                        placeholder="Type your video idea or feedback here (any language)...",
+                        lines=2,
+                        label="Your message"
+                    )
+                    send_btn = gr.Button("💬 Send", variant="primary")
+                
+                with gr.Row():
+                    # Action buttons
+                    regenerate_btn = gr.Button("🔄 Regenerate")
+                    clear_btn = gr.Button("🗑️ Clear Chat")
+                    copy_last_btn = gr.Button("📋 Copy Last Prompt")
+                
+                # Debug info in collapsible section
+                with gr.Accordion("🔧 Debug Info", open=False):
+                    test_btn = gr.Button("🔬 Test Basic Generation", variant="secondary")
+                    test_output = gr.Textbox(
+                        label="Basic Generation Test",
+                        lines=2,
+                        interactive=False
+                    )
+                    debug_btn = gr.Button("Get Debug Info")
+                    debug_info = gr.Textbox(
+                        label="Debug Information",
+                        value="Click 'Get Debug Info' to see model status",
+                        lines=8,
+                        interactive=False
+                    )
 
             # Tab 4: Custom Builder
             with gr.Tab("🛠️ Custom Builder"):
@@ -905,7 +960,7 @@ def create_interface():
                     show_copy_button=True
                 )
 
-        # Event handlers
+        # Event handlers for Image Analysis tab
         analyze_btn.click(
             fn=analyze_image_with_zephyr,
             inputs=[image_input],
@@ -916,42 +971,86 @@ def create_interface():
             inputs=[scene_state],
             outputs=sample_prompts
         )
-        optimize_btn.click(
-            fn=optimize_user_prompt_with_zephyr,
-            inputs=[user_idea, scene_state],
-            outputs=[optimized_prompt, model_status]
+        
+        # Event handlers for AI Prompt Generator tab (Chat)
+        send_btn.click(
+            fn=chat_with_ai,
+            inputs=[user_message, chat_history_state, last_prompt_state, scene_state],
+            outputs=[user_message, chat_history_state, last_prompt_state]
         )
-        retry_btn.click(
-            fn=fallback_generate_prompt,
-            inputs=[user_idea, scene_state],
-            outputs=[optimized_prompt, model_status]
+        
+        # Also trigger on Enter key
+        user_message.submit(
+            fn=chat_with_ai,
+            inputs=[user_message, chat_history_state, last_prompt_state, scene_state],
+            outputs=[user_message, chat_history_state, last_prompt_state]
         )
+        
+        # Function to regenerate last prompt
+        def regenerate_last_prompt(chat_history, scene_info):
+            if not chat_history:
+                return chat_history, ""
+                
+            # Get the last user message
+            last_user_msg = chat_history[-1][0]
+            new_prompt, _ = optimize_user_prompt_with_zephyr(last_user_msg, scene_info)
+            
+            # Add regenerated response to chat
+            chat_history.append(["🔄 Regenerate", f"Here's an alternative version:\n\n**{new_prompt}**"])
+            return chat_history, new_prompt
+        
+        regenerate_btn.click(
+            fn=regenerate_last_prompt,
+            inputs=[chat_history_state, scene_state],
+            outputs=[chat_history_state, last_prompt_state]
+        )
+        
+        # Clear chat button
+        clear_btn.click(
+            fn=lambda: ([], ""),
+            inputs=[],
+            outputs=[chat_history_state, last_prompt_state]
+        )
+        
+        # Copy last prompt button
+        def copy_last_prompt(last_prompt):
+            if last_prompt:
+                return gr.update(value=f"Copied: {last_prompt[:20]}...")
+            return gr.update(value="No prompt to copy")
+            
+        copy_last_btn.click(
+            fn=copy_last_prompt, 
+            inputs=[last_prompt_state],
+            outputs=[test_output]
+        )
+        
+        # Diagnostic buttons
         test_btn.click(
             fn=test_basic_generation,
             inputs=[],
             outputs=[test_output]
         )
+        
         debug_btn.click(
             fn=get_debug_info,
             inputs=[],
             outputs=[debug_info]
         )
-        refine_btn.click(
-            fn=refine_prompt_with_zephyr,
-            inputs=[optimized_prompt, feedback_input, chat_history_state, scene_state],
-            outputs=[optimized_prompt, chat_history_state]
-        )
+        
         # Update chat display when history changes
         chat_history_state.change(
             fn=lambda history: history,
             inputs=[chat_history_state],
             outputs=[chat_display]
         )
+        
+        # Event handlers for Custom Builder tab
         build_custom_btn.click(
             fn=build_custom_prompt,
             inputs=[custom_foundation, subject_motion, scene_motion, camera_motion, style_motion],
             outputs=[custom_output]
         )
+        
     return demo
 
 # Launch the app