Update server.py

server.py

@@ -81,7 +81,7 @@ def extract_segments_from_response(response_text: str) -> List[Dict]:
                             "segment_number": item.get("segment_number", len(segments) + 1),
                             "title": item.get("title", f"Segment {len(segments) + 1}"),
                             "start_time": parse_segment_timestamp(item.get("start_time", "00:00:00")),
-                            "end_time": parse_segment_timestamp(item.get("end_time", "00:01:00")),
+                            "end_time": parse_segment_timestamp(item.get("end_time", "00:10:00")),  # 10 minutes
                             "description": item.get("description", ""),
                             "engagement_level": item.get("engagement_level", "high"),
                             "reason": item.get("reason", "")
@@ -108,7 +108,7 @@ def extract_segments_from_response(response_text: str) -> List[Dict]:
                 end_time = f"{int(time_match.group(4)):02d}:{time_match.group(5)}:{time_match.group(6)}"
             else:
                 start_time = "00:00:00"
-                end_time = "00:01:00"
+                end_time = "00:10:00"  # 10 minutes default
             
             # Extract first sentence as title
             title_match = re.match(r'([^.\n]+)', part.strip())
@@ -152,21 +152,24 @@ async def process_transcription_for_highlights(
         system_prompt = """You are a movie marketing expert who identifies the most engaging and thrilling segments of movies.
 You will receive a full movie transcript with timestamps. Your task is to identify exactly 10 of the most compelling moments that would make audiences want to watch the full movie.
 
-IMPORTANT: You MUST respond with a valid JSON array. Do not include any text before or after the JSON array.
+IMPORTANT: 
+- You MUST respond with a valid JSON array ONLY. Do not include any text before or after the JSON array.
+- Each segment should be approximately 10 MINUTES LONG (600 seconds)
+- End time = Start time + ~10 minutes
 
 Each segment must have:
 - segment_number: (1-10)
 - title: (engaging, compelling title for this moment)
 - start_time: (HH:MM:SS format - when this segment starts)
-- end_time: (HH:MM:SS format - when this segment ends)
+- end_time: (HH:MM:SS format - when this segment ends, approximately 10 minutes after start)
 - description: (brief description of why this is engaging)
 - engagement_level: (high/medium)
 - reason: (one-line reason this will hook viewers)
 
 Return ONLY the JSON array. Example format:
 [
-    {"segment_number": 1, "title": "Epic Action Scene", "start_time": "00:15:32", "end_time": "00:18:45", "description": "...", "engagement_level": "high", "reason": "..."},
-    {"segment_number": 2, "title": "Emotional Climax", "start_time": "00:45:12", "end_time": "00:48:30", "description": "...", "engagement_level": "high", "reason": "..."}
+    {"segment_number": 1, "title": "Epic Action Scene", "start_time": "00:15:30", "end_time": "00:25:30", "description": "...", "engagement_level": "high", "reason": "..."},
+    {"segment_number": 2, "title": "Emotional Climax", "start_time": "00:45:12", "end_time": "00:55:12", "description": "...", "engagement_level": "high", "reason": "..."}
 ]
 """
         
@@ -366,4 +369,4 @@ async def get_status():
 if __name__ == "__main__":
     print("Starting Movie Highlight Extraction Service on port 7861...")
     print("Will automatically scan and process transcriptions on startup")
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7861)