Update app.py

app.py

@@ -103,6 +103,7 @@ def get_completion(
     top_p,
     temperature,
     model_choice,
+    conversation_history=None,
     previous_error=None,
     previous_command=None,
 ):
@@ -164,10 +165,12 @@ FORMAT DETECTION KEYWORDS:
 
     user_content += "\n\nYOUR RESPONSE:"
 
-    messages = [
-        {
-            "role": "system",
-            "content": """
+    # Initialize conversation with system message and first user message
+    if conversation_history is None:
+        messages = [
+            {
+                "role": "system",
+                "content": """
 You are a very experienced media engineer, controlling a UNIX terminal.
 You are an FFMPEG expert with years of experience and multiple contributions to the FFMPEG project.
 
@@ -224,12 +227,53 @@ When creating slideshows from multiple images with different dimensions, ALWAYS
 
 Remember: Simpler is better. Only use advanced ffmpeg features if absolutely necessary for the requested output.
 """,
-        },
-        {
-            "role": "user",
-            "content": user_content,
-        },
-    ]
+            },
+            {
+                "role": "user",
+                "content": user_content,
+            },
+        ]
+    else:
+        # Use existing conversation history
+        messages = conversation_history[:]
+        
+        # If there's a previous error, add it as a separate message exchange
+        if previous_error and previous_command:
+            # Add the failed command as assistant response
+            messages.append({
+                "role": "assistant", 
+                "content": f"I'll execute this FFmpeg command:\n\n```bash\n{previous_command}\n```"
+            })
+            
+            # Add the error as user feedback
+            messages.append({
+                "role": "user",
+                "content": f"""The command failed with the following error:
+
+ERROR MESSAGE:
+{previous_error}
+
+Please analyze the error and generate a corrected command that addresses the specific issue.
+
+COMMON SLIDESHOW ERROR FIXES:
+- If you see "do not match the corresponding output link" → Images have different dimensions, use scale+pad approach
+- If you see "Padded dimensions cannot be smaller than input dimensions" → Fix pad calculation or use standard resolution (1920x1080 or 1080x1920)
+- If you see "Failed to configure input pad" → Check scale and pad syntax, ensure proper filter chain
+- If you see "Invalid argument" in filters → Simplify filter_complex syntax and check parentheses
+
+FORMAT DETECTION KEYWORDS:
+- "vertical", "portrait", "9:16", "TikTok", "Instagram Stories", "phone" → Use 1080x1920
+- "horizontal", "landscape", "16:9", "YouTube", "TV" → Use 1920x1080 (default)
+- "square", "1:1", "Instagram post" → Use 1080x1080
+
+Please provide a corrected FFmpeg command."""
+            })
+        else:
+            # Add new user request to existing conversation
+            messages.append({
+                "role": "user",
+                "content": user_content,
+            })
     try:
         # Print the complete prompt
         print("\n=== COMPLETE PROMPT ===")
@@ -303,7 +347,13 @@ Remember: Simpler is better. Only use advanced ffmpeg features if absolutely nec
         # remove output.mp4 with the actual output file path
         command = command.replace("output.mp4", "")
 
-        return command
+        # Add the assistant's response to conversation history
+        messages.append({
+            "role": "assistant",
+            "content": content
+        })
+
+        return command, messages
     except Exception as e:
         raise Exception("API Error")
 
@@ -371,16 +421,18 @@ def update(
     command_attempts = []
     previous_error = None
     previous_command = None
+    conversation_history = None
 
     while attempts < 2:
         print("ATTEMPT", attempts + 1)
         try:
-            command_string = get_completion(
+            command_string, conversation_history = get_completion(
                 prompt,
                 files_info,
                 top_p,
                 temperature,
                 model_choice,
+                conversation_history,
                 previous_error,
                 previous_command,
             )