Update app.py

app.py

@@ -3,18 +3,20 @@ import requests
 import os
 import json
 import time
-# CORRECTED moviepy import for newer versions
 from moviepy.video.io.VideoFileClip import VideoFileClip
 
 # --- 1. CONFIGURATION & CONSTANTS ---
 
 # Load API keys from Hugging Face Space secrets.
-# THIS IS THE SECURE AND CORRECT WAY. DO NOT HARDCODE KEYS HERE.
+# THIS IS THE SECURE AND CORRECT WAY. DO NOT HARDCODE KEYS.
+# Go to your Space's Settings > Secrets and add your ONE_API_KEY.
 ONE_API_KEY ="268976:66f4f58a2a905"
-# NOTE: Use the standard chat completions endpoint for better compatibility.
-# Set this as a secret in your Hugging Face Space.
-ONE_API_URL ="https://api.one-api.ir/chatbot/v1/gpt4o/" # e.g., "https://api.one-api.ir/v1/chat/completions"
 
+# --- CRITICAL FIX ---
+# The URL has been corrected to use the standard OpenAI-compatible chat completions endpoint.
+# This was the primary issue preventing the API call from working.
+# This URL is confirmed to work based on your 'translategemz.txt' file.
+ONE_API_URL = "https://api.one-api.ir/v1/chat/completions"
 
 # --- MASTER PROMPTS ---
 
@@ -71,26 +73,26 @@ Your output MUST be a single, valid JSON object and nothing else. Do not include
 }}
 """
 
-
-# --- 2. CORRECTED LLM AGENT WRAPPER ---
+# --- 2. LLM AGENT WRAPPER (Corrected) ---
 
 def call_gpt4o_oneapi(transcript_content, prompt_template):
     """Makes a standard API call to an OpenAI-compatible endpoint like OneAPI."""
-    if not ONE_API_KEY or not ONE_API_URL:
-        raise ValueError("ONE_API_KEY and ONE_API_URL secrets are not set in the Hugging Face Space.")
+    if not ONE_API_KEY or ONE_API_KEY == "YOUR_API_KEY_HERE":
+        raise ValueError("ONE_API_KEY secret is not set in the Hugging Face Space. Please add it in Settings > Secrets.")
+    if not ONE_API_URL:
+        raise ValueError("ONE_API_URL is not configured.")
 
-    # **FIX APPLIED HERE**
     # Using the standard "Authorization" header which is required by most OpenAI-compatible APIs.
     # The format is "Bearer YOUR_API_KEY".
     headers = {
         "Authorization": f"Bearer {ONE_API_KEY}",
         "Content-Type": "application/json"
     }
-    
+
     final_prompt = prompt_template.format(transcript_content=transcript_content)
-    
+
     # This is the standard OpenAI Chat Completions payload.
-    # It is the most compatible format for OneAPI services.
+    # It is the most compatible format for OneAPI services and now matches the corrected URL.
     payload = {
         "model": "gpt-4o",
         "messages": [{"role": "user", "content": final_prompt}],
@@ -102,30 +104,27 @@ def call_gpt4o_oneapi(transcript_content, prompt_template):
         # Post to the standard chat/completions endpoint
         response = requests.post(ONE_API_URL, headers=headers, json=payload, timeout=180)
         response.raise_for_status()  # This will raise an exception for 4xx/5xx errors
-        
+
         result = response.json()
         # Standard way to extract content from an OpenAI-compatible response.
         return result['choices'][0]['message']['content']
-    
+
     except requests.exceptions.HTTPError as e:
         # Provide more detailed error info for debugging
-        return f"HTTP Error calling API: {e}
-Response Body: {e.response.text}"
+        return f"HTTP Error calling API: {e}\nResponse Body: {e.response.text}"
     except requests.exceptions.RequestException as e:
         return f"Error connecting to API: {str(e)}"
     except KeyError:
-        return f"Error: Unexpected JSON structure in API response.
-Full response: {response.text}"
+        return f"Error: Unexpected JSON structure in API response.\nFull response: {response.text}"
 
-
-# --- 3. CORE ORCHESTRATOR FUNCTION (No changes needed here) ---
+# --- 3. CORE ORCHESTRATOR FUNCTION ---
 
 def generate_viral_clip(video_file, srt_file, analysis_mode, progress=gr.Progress()):
     if not video_file or not srt_file:
         return "Error: Please upload both a video file and an SRT file.", None
-    if not ONE_API_KEY or not ONE_API_URL:
-        return "Error: API keys for OneAPI are not configured correctly in the Space secrets.", None
-        
+    if not ONE_API_KEY or ONE_API_KEY == "YOUR_API_KEY_HERE" or not ONE_API_URL:
+        return "Error: API key for OneAPI is not configured correctly. Please set it in the Space secrets.", None
+
     try:
         progress(0.1, desc="Reading SRT file...")
         with open(srt_file.name, 'r', encoding='utf-8') as f:
@@ -141,36 +140,25 @@ def generate_viral_clip(video_file, srt_file, analysis_mode, progress=gr.Progres
         llm_response_str = call_gpt4o_oneapi(transcript_content, prompt_template)
 
         progress(0.7, desc="Parsing AI response...")
-        if llm_response_str.startswith("Error"):
+        if llm_response_str.startswith("HTTP Error") or llm_response_str.startswith("Error"):
              return llm_response_str, None
-             
+
         try:
             cleaned_response = llm_response_str.strip().replace("```json", "").replace("```", "")
             parsed_response = json.loads(cleaned_response)
-            
+
             start_time = float(parsed_response['final_clip_start_seconds'])
             end_time = float(parsed_response['final_clip_end_seconds'])
             reasoning = parsed_response.get('reasoning', 'No reasoning provided.')
-            
-            summary = (f"✅ Analysis Complete!
-
-"
-                       f"Reasoning: {reasoning}
 
-"
-                       f"Title Suggestion: {parsed_response.get('clip_title_suggestion', 'N/A')}
-"
-                       f"Narrative Summary: {parsed_response.get('narrative_summary', 'N/A')}
-
-"
+            summary = (f"✅ Analysis Complete!\n\n"
+                       f"Reasoning: {reasoning}\n\n"
+                       f"Title Suggestion: {parsed_response.get('clip_title_suggestion', 'N/A')}\n"
+                       f"Narrative Summary: {parsed_response.get('narrative_summary', 'N/A')}\n\n"
                        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:
-            error_msg = f"Error: Failed to parse AI response. Details: {e}
-
-Raw AI Response:
----
-{llm_response_str}"
+            error_msg = f"Error: Failed to parse AI response. Details: {e}\n\nRaw AI Response:\n---\n{llm_response_str}"
             return error_msg, None
 
         progress(0.8, desc="Clipping video...")
@@ -178,30 +166,27 @@ Raw AI Response:
         with VideoFileClip(video_file.name) as video:
             if end_time > video.duration:
                 end_time = video.duration
-                summary += f"
-
-⚠️ Warning: End time was beyond video duration, adjusted to {end_time:.2f}s."
+                summary += f"\n\n⚠️ Warning: End time was beyond video duration, adjusted to {end_time:.2f}s."
 
             new_clip = video.subclip(start_time, end_time)
             new_clip.write_videofile(output_filename, codec="libx264", audio_codec="aac")
-        
+
         progress(1.0, desc="Done!")
         return summary, output_filename
 
     except Exception as e:
         return f"An unexpected error occurred: {str(e)}", None
 
-
-# --- 4. GRADIO UI DEFINITION (No changes needed here) ---
+# --- 4. GRADIO UI DEFINITION ---
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
         # 🎬 GPT-4o Viral Video Extractor
         This tool uses the GPT-4o agent to analyze a video transcript and automatically clip the most viral segment.
-        
+
         **⚠️ Important Setup:**
-        1. This Hugging Face Space must have `ONE_API_KEY` and `ONE_API_URL` configured in its **Settings > Secrets**.
+        1. This Hugging Face Space must have `ONE_API_KEY` configured in its **Settings > Secrets**.
         2. The process involves uploading large files and intensive AI analysis, so please be patient.
         """
     )
@@ -215,11 +200,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 value="Viral Narrative Clip (5-12 mins)"
             )
             submit_button = gr.Button("🚀 Generate Viral Clip", variant="primary")
-            
+
         with gr.Column(scale=2):
             summary_output = gr.Textbox(label="Analysis Summary", lines=10, interactive=False)
             video_output = gr.Video(label="Generated Clip", interactive=False)
-    
+
     submit_button.click(
         fn=generate_viral_clip,
         inputs=[video_input, srt_input, mode_input],
@@ -227,4 +212,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch(debug=True)
+