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)