apapis added

scripts/generate_summary.py

@@ -0,0 +1,42 @@
+import anthropic
+import os
+from dotenv import load_dotenv
+import json
+from src.prompt import general_summary_prompt,custom_template_prompt
+load_dotenv()
+
+def general_summary(transcription):
+    client = anthropic.Anthropic(
+        api_key=os.getenv("ANTHTROPIC_API_KEY"),
+    )
+    message = client.messages.create(
+        model="claude-3-5-sonnet-20241022",
+        max_tokens=4000,
+        messages=[
+            {"role": "user", "content": f"{general_summary_prompt}"},
+            {"role": "user", "content": f"Transcription: {transcription}"}
+        ]
+    )
+
+    text = message.content[0].text
+    return json.loads(text)
+
+
+
+
+def custom_summary(template, transcription):
+    client = anthropic.Anthropic(
+        api_key=os.getenv("ANTHTROPIC_API_KEY"),
+    )
+    message = client.messages.create(
+        model="claude-3-5-sonnet-20241022",
+        max_tokens=8000,
+        messages=[
+            {"role": "user", "content": f"{custom_template_prompt}"},
+            {"role": "user", "content": f"TEMPLATE : {template}"},
+            {"role": "user", "content": f"Transcription: {transcription}"}
+        ]
+    )
+
+    text = message.content[0].text
+    return json.loads(text)

scripts/transcriber.py

@@ -123,7 +123,7 @@ def transcribe_media(file_loc: str, media_type: str = "audio"):
         else:
             raise ValueError("media_type must be either 'audio' or 'video'.")
 
-        print(f"Transcription response: {response}\n\n")
+       
         return response
 
     except Exception as e: