Update README.md

README.md

@@ -80,6 +80,69 @@ PARAMETER stop "<s>"
 PARAMETER stop "</s>"
 ```
 
+## 💻 Ollama Python Test Code
+
+```
+import sys
+import os
+import requests
+from bs4 import BeautifulSoup
+from langchain_community.chat_models import ChatOllama
+from langchain.schema import AIMessage, HumanMessage, SystemMessage
+
+def clean_output(text):
+    text = text.replace("</s>", "").strip()
+    return text
+
+def invoke_model(text):
+    messages = [
+        SystemMessage(content='You are an expert copywriter with expertise in summarizing documents.'),
+        HumanMessage(content=f'Please provide a short and concise summary of the following text:\nTEXT: {text}')
+    ]
+    
+    try:
+        llm = ChatOllama(model="smartllama-3-Ko-8b-256k-pose:latest")
+        summary_output = llm.invoke(messages)
+        if isinstance(summary_output, AIMessage):
+            cleaned_content = clean_output(summary_output.content)
+            return cleaned_content
+        else:
+            return "Unexpected data type for model output."
+    except Exception as e:
+        print(f"An error occurred while processing the model output: {str(e)}")
+        return None
+
+def fetch_text_from_url(url):
+    try:
+        response = requests.get(url)
+        response.raise_for_status()  # Raises an HTTPError for bad responses
+        soup = BeautifulSoup(response.text, 'html.parser')
+        # Extract main content from paragraphs within the article body only
+        content = soup.find('div', {'id': 'bodyContent'})
+        paragraphs = content.find_all('p')
+        text_content = ' '.join(p.text for p in paragraphs)
+        return text_content
+    except requests.RequestException as e:
+        print(f"Failed to fetch data from URL: {str(e)}")
+        return None
+
+def summarize_content(source):
+    text_content = fetch_text_from_url(source) if source.startswith(('http://', 'https://')) else read_text_file(source)
+    if text_content:
+        summary = invoke_model(text_content)
+        print("Summary of the document:")
+        print(summary)
+    else:
+        print("No text found or unable to extract text from source.")
+
+if __name__ == '__main__':
+    if len(sys.argv) < 2:
+        print("Usage: python script.py <file_path_or_url>")
+    else:
+        source = sys.argv[1]
+        summarize_content(source)
+```
+
 ### 🗞️ Configuration
 The YAML configuration for this model: