Spaces:

Hasitha16
/

churnsight-ai

Running

App Files Files Community

Hasitha16 commited on Jul 13

Commit

57d0c46

verified ·

1 Parent(s): 3b153d2

Update main.py

Browse files

Files changed (1) hide show

main.py +11 -98

main.py CHANGED Viewed

@@ -7,7 +7,6 @@ from pydantic import BaseModel
 from transformers import pipeline
 from io import StringIO
 import os, csv, logging
-from openai import OpenAI
 from model import summarize_review, smart_summarize
 from typing import Optional
@@ -40,71 +39,7 @@ def custom_swagger_ui():
 @app.get("/", response_class=HTMLResponse)
 def root():
-    return """
-    <html>
-    <head>
-        <title>NeuroPulse AI</title>
-        <style>
-            body {
-                font-family: 'Segoe UI', sans-serif;
-                background: linear-gradient(135deg, #f0f4ff, #fef3c7);
-                margin: 0;
-                padding: 60px;
-                text-align: center;
-                color: #1f2937;
-            }
-            .container {
-                background: white;
-                padding: 40px;
-                border-radius: 16px;
-                max-width: 800px;
-                margin: auto;
-                box-shadow: 0 10px 30px rgba(0,0,0,0.08);
-                animation: fadeIn 1s ease-in-out;
-            }
-            @keyframes fadeIn {
-                from {opacity: 0; transform: translateY(20px);}
-                to {opacity: 1; transform: translateY(0);}
-            }
-            h1 {
-                font-size: 36px;
-                margin-bottom: 12px;
-                color: #4f46e5;
-            }
-            p {
-                font-size: 18px;
-                margin-bottom: 32px;
-            }
-            .btn {
-                display: inline-block;
-                margin: 8px;
-                padding: 14px 24px;
-                border-radius: 8px;
-                font-weight: 600;
-                color: white;
-                text-decoration: none;
-                background: linear-gradient(90deg, #4f46e5, #6366f1);
-                transition: all 0.3s ease;
-            }
-            .btn:hover {
-                transform: translateY(-2px);
-                box-shadow: 0 4px 12px rgba(0,0,0,0.1);
-            }
-            .btn.red {
-                background: linear-gradient(90deg, #dc2626, #ef4444);
-            }
-        </style>
-    </head>
-    <body>
-        <div class="container">
-            <h1>🧠 Welcome to <strong>NeuroPulse AI</strong></h1>
-            <p>Smarter AI feedback analysis — Summarization, Sentiment, Emotion, Aspects, LLM Q&A, and Metadata Tags.</p>
-            <a class="btn" href="/docs">📘 Swagger UI</a>
-            <a class="btn red" href="/redoc">📕 ReDoc</a>
-        </div>
-    </body>
-    </html>
-    """
 class ReviewInput(BaseModel):
     text: str
@@ -114,6 +49,9 @@ class ReviewInput(BaseModel):
     follow_up: Optional[str] = None
     product_category: Optional[str] = None
     device: Optional[str] = None
 class BulkReviewInput(BaseModel):
     reviews: list[str]
@@ -123,14 +61,6 @@ class BulkReviewInput(BaseModel):
     product_category: Optional[list[str]] = None
     device: Optional[list[str]] = None
-class ChatInput(BaseModel):
-    question: str
-    context: str
-class TranslationInput(BaseModel):
-    text: str
-    target_lang: str = "fr"
 VALID_API_KEY = "my-secret-key"
 logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
@@ -178,7 +108,7 @@ async def bulk(data: BulkReviewInput, x_api_key: str = Header(None)):
     return {"results": results}
 @app.post("/analyze/")
-async def analyze(request: Request, data: ReviewInput, x_api_key: str = Header(None), download: str = None):
     if x_api_key != VALID_API_KEY:
         raise HTTPException(status_code=401, detail="Invalid or missing API key")
@@ -203,7 +133,11 @@ async def analyze(request: Request, data: ReviewInput, x_api_key: str = Header(N
                     "score": asp_result["score"]
                 })
-    follow_up_response = chat_llm(data.follow_up, data.text) if data.follow_up else None
     return {
         "summary": summary,
@@ -211,33 +145,12 @@ async def analyze(request: Request, data: ReviewInput, x_api_key: str = Header(N
         "emotion": emotion,
         "aspects": aspects_list,
         "follow_up": follow_up_response,
         "product_category": auto_fill(data.product_category),
         "device": auto_fill(data.device, "Web"),
         "industry": auto_fill(data.industry)
     }
-@app.post("/translate/")
-async def translate(data: TranslationInput):
-    translator = pipeline("translation", model=f"Helsinki-NLP/opus-mt-en-{data.target_lang}")
-    return {"translated_text": translator(data.text)[0]["translation_text"]}
-@app.post("/chat/")
-async def chat(input: ChatInput, x_api_key: str = Header(None)):
-    if x_api_key != VALID_API_KEY:
-        raise HTTPException(status_code=401, detail="Invalid or missing API key")
-    return {"response": chat_llm(input.question, input.context)}
-def chat_llm(question, context):
-    client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-    res = client.chat.completions.create(
-        model="gpt-3.5-turbo",
-        messages=[
-            {"role": "system", "content": "You are a helpful AI review analyst."},
-            {"role": "user", "content": f"Context: {context}\nQuestion: {question}"}
-        ]
-    )
-    return res.choices[0].message.content.strip()
 def custom_openapi():
     if app.openapi_schema:
         return app.openapi_schema

 from transformers import pipeline
 from io import StringIO
 import os, csv, logging
 from model import summarize_review, smart_summarize
 from typing import Optional
 @app.get("/", response_class=HTMLResponse)
 def root():
+    return open("app/static/index.html").read()
 class ReviewInput(BaseModel):
     text: str
     follow_up: Optional[str] = None
     product_category: Optional[str] = None
     device: Optional[str] = None
+    intelligence: Optional[bool] = False
+    verbosity: Optional[str] = "detailed"
+    explain: Optional[bool] = False
 class BulkReviewInput(BaseModel):
     reviews: list[str]
     product_category: Optional[list[str]] = None
     device: Optional[list[str]] = None
 VALID_API_KEY = "my-secret-key"
 logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
     return {"results": results}
 @app.post("/analyze/")
+async def analyze(request: Request, data: ReviewInput, x_api_key: str = Header(None)):
     if x_api_key != VALID_API_KEY:
         raise HTTPException(status_code=401, detail="Invalid or missing API key")
                     "score": asp_result["score"]
                 })
+    follow_up_response = None
+    if data.follow_up and data.intelligence:
+        follow_up_response = f"[Mocked Detailed Answer in {data.verbosity} mode]"
+    explanation = "This summary was generated using transformer-based sequence modeling and contextual keyword expansion." if data.explain else None
     return {
         "summary": summary,
         "emotion": emotion,
         "aspects": aspects_list,
         "follow_up": follow_up_response,
+        "explanation": explanation,
         "product_category": auto_fill(data.product_category),
         "device": auto_fill(data.device, "Web"),
         "industry": auto_fill(data.industry)
     }
 def custom_openapi():
     if app.openapi_schema:
         return app.openapi_schema