temporaru

Author: kwisznie

README.md

@@ -29,7 +29,7 @@ A magical AI that turns your boring long documents into... slightly less boring
 
 It doesn't work, we used the wrong AI :( and Azure for Students doesn't let us access the OpenAI model.
 
-We can only use a special model to summarize text files, but this Azure service just copies and pastes random sentences from our text. 
+We can only use a special model to summarize text files, but this Azure service just copies and pastes random sentences from our text.
 ```ini
 AZURE_ENDPOINT='https://kaka-doc-summary-ai.cognitiveservices.azure.com/'
 AZURE_API_KEY='your-magic-code-here'
@@ -61,3 +61,12 @@ streamlit run app/main.py       # Do the thing
 - Azure for the free(ish) toys
 - My cat for moral support 🐈
 - Coffee ☕ (the real MVP)
+
+
+# Nowego gowno
+## env file
+```ini
+API_KEY='api'
+BASE_URL='https://openrouter.ai/api/v1'
+MODEL='mistralai/mistral-small-3.1-24b-instruct:free'
+```

app/src/summarizer.py

@@ -1,46 +1,31 @@
-from azure.core.credentials import AzureKeyCredential
-from azure.ai.textanalytics import (
-    TextAnalyticsClient,
-    ExtractiveSummaryAction
-)
+from openai import OpenAI
 from config.settings import Settings
 
 class Summarizer:
     def __init__(self, config: Settings):
-        self.endpoint = config.AZURE_ENDPOINT
-        self.key = config.AZURE_API_KEY
-        self.client = TextAnalyticsClient(
-            endpoint=self.endpoint,
-            credential=AzureKeyCredential(self.key)
+        self.api_key = config.API_KEY
+        self.base_url = config.BASE_URL
+        self.model = config.MODEL
+        self.client = OpenAI(
+            base_url=self.base_url,
+            api_key=self.api_key,
         )
 
     def summarize(self, text: str) -> str:
         try:
-            actions = [
-                ExtractiveSummaryAction(
-                    max_sentence_count=15,
-                    order_by="Rank"
-                )
-            ]
-
-            poller = self.client.begin_analyze_actions(
-                documents=[text],
-                actions=actions,
-                language="pl",
-                show_stats=True
+            completion = self.client.chat.completions.create(
+                model = self.model,
+                messages=[
+                    {
+                        "role": "user",
+                        "content": f"Summarize this text in the same language:\n\n{text}"
+                    }
+                ]
             )
 
-            results = list(poller.result())
-
-            summary_sentences = []
-            for result in results:
-                for action_result in result:
-                    if not action_result.is_error:
-                        summary_sentences.extend(
-                            sentence.text for sentence in action_result.sentences
-                        )
+            result = completion.choices[0].message.content
 
-            return " ".join(summary_sentences) if summary_sentences else "No summary generated"
+            return result
 
         except Exception as e:
             raise Exception(f"Summarization error: {str(e)}")

config/settings.py

@@ -5,9 +5,8 @@
 class Settings:
     def __init__(self):
         load_dotenv(dotenv_path=Path(".env"))
-        self.AZURE_ENDPOINT = os.getenv("AZURE_ENDPOINT", "").strip()
-        self.AZURE_API_KEY = os.getenv("AZURE_API_KEY", "")
-        self.API_VERSION = os.getenv("API_VERSION", "2023-05-15")
-        self.MAX_TOKENS = int(os.getenv("MAX_TOKENS", "1"))
+        self.API_KEY = os.getenv("API_KEY", "")
+        self.BASE_URL = os.getenv("BASE_URL", "")
+        self.MODEL = os.getenv("MODEL", "")
 
 settings = Settings()

requirements.txt

@@ -4,5 +4,4 @@ dotenv
 requests
 typing
 pytest
-azure-ai-textanalytics>=5.3.0
-azure-core>=1.28.0
+openai