Update app.py

app.py

@@ -6,16 +6,12 @@ import time
 from moviepy.video.io.VideoFileClip import VideoFileClip
 
 # --- 1. CONFIGURATION & CONSTANTS ---
-
 # Securely load API key from Hugging Face Space secrets.
-ONE_API_KEY = os.environ.get("ONE_API_KEY", "268976:66f4f58a2a905") # Using your key for now, but secrets are better.
-
+ONE_API_KEY = os.environ.get("ONE_API_KEY", "268976:66f4f58a2a905") # Using your key for now, secrets are better.
 # The custom endpoint for the one-api.ir service.
 ONE_API_URL = "https://api.one-api.ir/chatbot/v1/gpt4o/"
 
 # --- MASTER PROMPTS ---
-
-# Prompt for Mode 1: "Viral Spot for Shorts (< 3 mins)"
 PROMPT_SHORTS_MODE = """
 You are an expert producer of viral short-form content. Analyze the provided SRT transcript to find the single most impactful, hook-worthy segment for a video under 3 minutes (ideally 60-90 seconds).
 
@@ -35,7 +31,6 @@ Your output MUST be a single, valid JSON object and nothing else. Do not include
 }}
 """
 
-# Prompt for Mode 2: "Viral Narrative Clip (5-12 mins)"
 PROMPT_NARRATIVE_MODE = """
 You are an expert video editor and storyteller. Analyze the provided transcript to find the most compelling narrative segment between 5 and 12 minutes long.
 
@@ -55,10 +50,10 @@ Your output MUST be a single, valid JSON object and nothing else. Do not include
 }}
 """
 
-# --- 2. LLM AGENT WRAPPER ---
+# --- 2. LLM AGENT WRAPPER (WITH DEFINITIVE FIX) ---
 
 def call_gpt4o_oneapi(transcript_content, prompt_template):
-    """Makes a custom API call to the one-api.ir service."""
+    """Makes a robust, custom API call to the one-api.ir service."""
     if not ONE_API_KEY or not ONE_API_URL:
         raise ValueError("ONE_API_KEY and ONE_API_URL secrets are not set correctly.")
 
@@ -74,14 +69,27 @@ def call_gpt4o_oneapi(transcript_content, prompt_template):
         response.raise_for_status()
         result = response.json()
         
-        if "result" in result and isinstance(result.get("result"), list) and len(result["result"]) > 0:
-            message_content = result["result"][0].get("content")
+        # Check for the basic expected structure
+        if "result" not in result or not isinstance(result.get("result"), list) or len(result["result"]) == 0:
+            return f"Error: Unexpected JSON structure from API. 'result' list not found or empty.\nFull response: {json.dumps(result)}"
+
+        first_item_in_result = result["result"][0]
+
+        # --- THE CRITICAL FIX IS HERE ---
+        # The API can return a dict OR a string inside the list. We must handle both.
+        if isinstance(first_item_in_result, dict):
+            # This is the expected case: {"content": "..."}
+            message_content = first_item_in_result.get("content")
             if message_content:
                 return message_content
             else:
-                 return f"Error: 'content' key not found in API response item.\nFull response: {json.dumps(result)}"
+                 return f"Error: 'content' key not found in API response dictionary.\nFull response: {json.dumps(result)}"
+        elif isinstance(first_item_in_result, str):
+            # This is the alternate case: the string is the entire response.
+            return first_item_in_result
         else:
-            return f"Error: Unexpected JSON structure in API response. 'result' list not found or empty.\nFull response: {json.dumps(result)}"
+            # Handle any other unexpected type
+            return f"Error: Unknown item type in API 'result' list.\nFull response: {json.dumps(result)}"
 
     except requests.exceptions.HTTPError as e:
         return f"HTTP Error calling API: {e}\nResponse Body: {e.response.text}"
@@ -114,12 +122,15 @@ def generate_viral_clip(video_file, srt_file, analysis_mode, progress=gr.Progres
              return llm_response_str, None
 
         try:
-            cleaned_response = llm_response_str.strip().replace("```json", "").replace("```", "")
+            # Clean potential markdown code fences from the AI response
+            cleaned_response = llm_response_str.strip()
+            if cleaned_response.startswith("```json"):
+                cleaned_response = cleaned_response[7:]
+            if cleaned_response.endswith("```"):
+                cleaned_response = cleaned_response[:-3]
+            
             parsed_response = json.loads(cleaned_response)
 
-            # --- ROBUSTNESS FIX APPLIED HERE ---
-            # Check if the parsed response is a dictionary. If not, the AI returned a simple string
-            # instead of the requested JSON object. Raise a TypeError to handle it as a parsing failure.
             if not isinstance(parsed_response, dict):
                 raise TypeError(f"AI did not return a valid JSON object. It returned a {type(parsed_response).__name__}.")
 
@@ -134,7 +145,6 @@ def generate_viral_clip(video_file, srt_file, analysis_mode, progress=gr.Progres
                        f"Clipping video from {time.strftime('%H:%M:%S', time.gmtime(start_time))} to {time.strftime('%H:%M:%S', time.gmtime(end_time))}.")
 
         except (json.JSONDecodeError, KeyError, TypeError) as e:
-            # The new TypeError will be caught here, providing a clear error message.
             error_msg = f"Error: Failed to parse AI response. Details: {e}\n\nRaw AI Response:\n---\n{llm_response_str}"
             return error_msg, None
 
@@ -152,10 +162,10 @@ def generate_viral_clip(video_file, srt_file, analysis_mode, progress=gr.Progres
         return summary, output_filename
 
     except Exception as e:
-        return f"An unexpected error occurred: {str(e)}", None
+        # This generic catch-all is our final safety net.
+        return f"An unexpected error occurred in the main process: {str(e)}", None
 
 # --- 4. GRADIO UI DEFINITION ---
-
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """