changes

app.py

@@ -17,9 +17,17 @@ import requests
 import io
 from sklearn.preprocessing import StandardScaler
 from collections import defaultdict
-
+from dotenv import load_dotenv
+load_dotenv()  # load all the environment variables
+from openai import OpenAI
 warnings.filterwarnings('ignore')
 
+client = OpenAI(
+  base_url = "https://integrate.api.nvidia.com/v1",
+  api_key = os.getenv("NVidea_Key")
+)
+
+
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -700,4 +708,22 @@ async def get_financial_recommendations(customer_id: str):
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail=f"Error processing request: {str(e)}"
-        )
+        )
+    
+@app.get("/contentcreation/{category}")
+async def contentcreation(category:str):
+    completion = client.chat.completions.create(
+    model="microsoft/phi-4-mini-instruct",
+    messages=[{"role":"user","content":"Generate some investment articles in 10 words. Don't show the thinking part in the output"}],
+    temperature=0.6,
+    top_p=0.7,
+    max_tokens=4096,
+    stream=True
+    )
+    full_response = ""
+    for chunk in completion:
+        if chunk.choices[0].delta.content is not None:
+            full_response += chunk.choices[0].delta.content
+        # Clean the complete response
+    cleaned_response = full_response.replace("<think>", "").replace("</think>", "").strip()
+    return(cleaned_response)