Update app.py

app.py

@@ -1,32 +1,62 @@
 import gradio as gr
 from gradio_client import Client
-import ast
-import re
 
 
-def transcription_to_segments(q):
-    client = Client(
-        "https://huggingface-projects-llama-2-13b-chat.hf.space/--replicas/bk259/")
-    prompt = "your a friendly assistant."
+def call_api(transcript):
+    client = Client("http://47.103.63.15:50085/")
+    prompt = """
+    INSTRUCTIONS:
+    Convert this Transcript into Segment Cards.
+
+    IMPORTANT:
+    - Maintain Language Consistency: Ensure that the language used in segment cards matches the transcription for clarity.
+    - Verify Duration Accuracy: Double-check that the stated duration of each segment card accurately reflects the content's length.
+    - Avoid Special Characters and Emojis: Refrain from using special characters or emojis in segment card descriptions for a clean and professional appearance.
+    - There should be exactly the same number of segment cards as the transcript has segments
+    - Match the Number of Segment Cards: Create a segment card for each segment in the transcription to align with the video content.
+    - Focus Solely on Segment Cards: Output only the segment cards without any additional text or commentary.
+    - Implement <card> Tags: Properly format each segment card by including <card> tags before and after the card content.
+    - Precision and Completeness: Craft each segment card with precision and ensure it contains complete and relevant information.
+    - Adhere to Output Format: Format your answers according to the specified Output Format.
+    - Consider YouTube Short Video Rating: Evaluate the content's quality in terms of suitability for YouTube Short videos when providing a rating.
+    - Make Sure the Title is no more than 50 characters
+    - Make Sure the Description is no more than 100 characters
+    - It is absolutly important to use the same language as the transript.
+
+    
+    CONTENT OF CARD:
+    {
+    Title: (max. 50 Characters)
+    Description: (summary of content max. 140 Characters)
+    Rating: 1-5
+    Duration: [00:00.000 - 00:00.000]
+    }
+    TRANSCRIPT: 
+    """
+    transcript = str(transcript)
     result = client.predict(
-        q,
-        prompt,
-        1024,
-        0.6,
+        prompt + transcript,
         0.9,
-        50,
-        1.2,
-        api_name="/chat"
+        1024,
+        api_name="/predict"
     )
-    print(result)
     return result
 
 
+def transcription_to_segments(transcript):
+    transcript = ast.literal_eval(transcript)
+    segment_cards_array = []
+    for segment in transcript:
+        result = call_api(segment)
+        segment_cards_array.append(result)
+    return segment_cards_array
+
+
 # Define the Gradio interface for transcription and segmentation
 interface = gr.Interface(
     fn=transcription_to_segments,
     inputs="text",
-    outputs="text"
+    outputs="textbox"
 )
 
 # Launch the Gradio interface