Spaces:

ganna217
/

quickbrief

Sleeping

App Files Files Community

ganna217 commited on Mar 13, 2025

Commit

14adcdb

0 Parent(s):

Initial commit with all files

Browse files

Files changed (5) hide show

Dockerfile +12 -0
Moon.PNG +0 -0
app.py +87 -0
index.html +258 -0
requirements.txt +4 -0

Dockerfile ADDED Viewed

	@@ -0,0 +1,12 @@

+FROM python:3.9-slim
+WORKDIR /app
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+COPY . .
+EXPOSE 7860
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

Moon.PNG ADDED Viewed

app.py ADDED Viewed

	@@ -0,0 +1,87 @@

+from fastapi import FastAPI, Request
+from fastapi.responses import HTMLResponse
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
+from transformers import pipeline
+import traceback
+app = FastAPI()
+# Enable CORS to allow POST requests
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+# Serve static files (like Moon.png) under /assets
+app.mount("/assets", StaticFiles(directory="."), name="static")
+# Serve the HTML page at the root
+@app.get("/", response_class=HTMLResponse)
+async def serve_html():
+    with open("index.html", "r", encoding="utf-8") as f:
+        return HTMLResponse(content=f.read())
+# Load pre-trained models from Hugging Face Hub
+try:
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+    translators = {
+        "fr": pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr"),
+        "de": pipeline("translation", model="Helsinki-NLP/opus-mt-en-de"),
+        "es": pipeline("translation", model="Helsinki-NLP/opus-mt-en-es"),
+        "ar": pipeline("translation", model="Helsinki-NLP/opus-mt-en-ar")
+    }
+except Exception as e:
+    print(f"Error loading models: {str(e)}")
+    raise e
+# Summarization endpoint
+@app.post("/summarize")
+async def summarize(request: Request):
+    try:
+        data = await request.json()
+        text = data["text"]
+        input_length = len(summarizer.tokenizer(text)["input_ids"])
+        max_length = max(30, input_length // 2)
+        summary = summarizer(text, max_length=max_length, min_length=30, do_sample=False)
+        return {"summary": summary[0]["summary_text"]}
+    except Exception as e:
+        print(f"Error in summarize: {str(e)}")
+        print(traceback.format_exc())
+        return {"error": f"Failed to summarize: {str(e)}"}
+# Translation endpoint
+@app.post("/translate")
+async def translate(request: Request):
+    try:
+        data = await request.json()
+        text = data["text"]
+        lang = data["lang"]
+        if lang not in translators:
+            return {"error": "Language not supported"}
+        # Perform translation
+        result = translators[lang](text)
+        print(f"Translation result for {lang}: {result}")  # Debugging
+        # Check if result is a list and has at least one item
+        if not isinstance(result, list) or len(result) == 0:
+            return {"error": "Translation failed: empty or invalid result"}
+        # Extract the translated text (correct key is "translation_text")
+        translation = result[0].get("translation_text")
+        if translation is None:
+            return {"error": "Translation failed: 'translation_text' not found in result"}
+        return {"translation": translation}
+    except Exception as e:
+        print(f"Error in translate: {str(e)}")
+        print(traceback.format_exc())
+        return {"error": f"Failed to translate: {str(e)}"}
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)

index.html ADDED Viewed

	@@ -0,0 +1,258 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>QuickBrief</title>
+    <link href="https://fonts.googleapis.com/css2?family=Orbitron:wght@400;700&family=Papyrus&display=swap" rel="stylesheet">
+    <style>
+        * {
+            box-sizing: border-box;
+        }
+        body {
+            font-family: Arial, sans-serif;
+            background-color: #000000;
+            color: #e0e0e0;
+            margin: 0;
+            padding: 20px;
+            min-height: 100vh;
+            display: flex;
+            justify-content: center;
+            align-items: center;
+        }
+        .container {
+            max-width: 1000px;
+            width: 100%;
+            margin: 0 auto;
+        }
+        .header {
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            margin-bottom: 4px;
+        }
+        .header img {
+            width: 60px;
+            height: auto;
+            margin-right: 15px;
+        }
+        .header h1 {
+            font-family: 'Papyrus', sans-serif;
+            font-size: 32px;
+            margin: 0;
+            color: #ffffff;
+            text-shadow: 2px 2px 5px rgba(107, 72, 255, 0.5);
+        }
+        .tagline {
+            font-family: 'Orbitron', sans-serif;
+            font-size: 16px;
+            font-weight: 700;
+            color: #b0b0b0;
+            text-align: center;
+            margin-bottom: 30px;
+            letter-spacing: 1px;
+        }
+        /* Summarization Section */
+        .summarize-section {
+            display: flex;
+            flex-wrap: wrap;
+            gap: 20px;
+            margin-bottom: 40px;
+        }
+        .summarize-section div {
+            flex: 1;
+            min-width: 300px;
+        }
+        .summarize-section label {
+            display: block;
+            font-size: 18px;
+            color: #ffffff;
+            margin-bottom: 10px;
+        }
+        textarea, .output {
+            width: 100%;
+            padding: 15px;
+            border-radius: 10px;
+            border: 2px solid #6b48ff;
+            font-size: 16px;
+            background-color: #1a1a1a;
+            color: #ffffff;
+            box-shadow: 0 0 10px rgba(107, 72, 255, 0.3);
+        }
+        textarea {
+            height: 200px;
+            resize: both;
+        }
+        .output {
+            height: 200px;
+        }
+        .translation-output {
+            height: 120px;
+        }
+        .button-container {
+            width: 100%;
+            text-align: left;
+            margin-top: 10px;
+        }
+        button {
+            background: linear-gradient(90deg, #6b48ff, #a73aff);
+            color: white;
+            padding: 12px 30px;
+            border: none;
+            border-radius: 25px;
+            cursor: pointer;
+            font-size: 16px;
+            transition: transform 0.2s, box-shadow 0.3s;
+            box-shadow: 0 4px 15px rgba(107, 72, 255, 0.4);
+        }
+        button:hover {
+            background: linear-gradient(90deg, #5a38e7, #9632e6);
+            transform: scale(1.05);
+            box-shadow: 0 6px 20px rgba(107, 72, 255, 0.6);
+        }
+        /* Translation Section */
+        .translate-section {
+            margin-top: 40px;
+            display: flex;
+            flex-wrap: wrap;
+            gap: 15px;
+            align-items: center;
+            margin-bottom: 20px;
+        }
+        .translate-section label {
+            font-size: 18px;
+            color: #ffffff;
+        }
+        select {
+            padding: 10px;
+            border-radius: 5px;
+            border: 2px solid #a73aff;
+            font-size: 16px;
+            background-color: #1a1a1a;
+            color: #ffffff;
+            cursor: pointer;
+            width: 200px;
+            appearance: none;
+            background-image: url('data:image/svg+xml;utf8,<svg fill="white" height="24" viewBox="0 0 24 24" width="24" xmlns="http://www.w3.org/2000/svg"><path d="M7 10l5 5 5-5z"/></svg>');
+            background-repeat: no-repeat;
+            background-position: right 10px center;
+        }
+        select:focus {
+            outline: none;
+            border-color: #6b48ff;
+        }
+        /* Responsive Design */
+        @media (max-width: 768px) {
+            .summarize-section {
+                flex-direction: column;
+            }
+            .summarize-section div {
+                min-width: 100%;
+            }
+            .translate-section {
+                flex-direction: column;
+                align-items: flex-start;
+            }
+            select {
+                width: 100%;
+            }
+            button {
+                width: 100%;
+            }
+        }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <div class="header">
+            <img src="/assets/Moon.png" alt="QuickBrief Logo">
+            <h1>QuickBrief</h1>
+        </div>
+        <p class="tagline">Simplify. Summarize. Translate. Understand.</p>
+        <!-- Summarization Section -->
+        <div class="summarize-section">
+            <div>
+                <label>Input Text</label>
+                <textarea id="inputText" placeholder="Enter your text here..."></textarea>
+                <div class="button-container">
+                    <button onclick="summarize()">SUMMARIZE</button>
+                </div>
+            </div>
+            <div>
+                <label>Summary</label>
+                <div id="summary" class="output"></div>
+            </div>
+        </div>
+        <!-- Translation Section -->
+        <div class="translate-section">
+            <label>Translate Summary</label>
+            <select id="language">
+                <option value="default">Select Language</option>
+                <option value="fr">French</option>
+                <option value="de">German</option>
+                <option value="es">Spanish</option>
+                <option value="ar">Arabic</option>
+            </select>
+            <button onclick="doTranslate()">TRANSLATE</button>
+        </div>
+        <div>
+            <label>Translation</label>
+            <div id="translation" class="output translation-output"></div>
+        </div>
+    </div>
+    <script>
+        console.log("Script loaded successfully");
+        async function summarize() {
+            const text = document.getElementById("inputText").value;
+            try {
+                const response = await fetch("/summarize", {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ text: text })
+                });
+                const data = await response.json();
+                console.log("Summarize response:", JSON.stringify(data, null, 2));
+                if (data.summary) {
+                    document.getElementById("summary").innerText = data.summary;
+                } else {
+                    document.getElementById("summary").innerText = "No summary returned: " + (data.error || "Unknown error");
+                }
+            } catch (error) {
+                console.error("Summarize error:", error);
+                document.getElementById("summary").innerText = "Failed to summarize: " + error.message;
+            }
+        }
+        async function doTranslate() {
+            const summary = document.getElementById("summary").innerText;
+            const lang = document.getElementById("language").value;
+            if (lang === "default" || summary === "") {
+                document.getElementById("translation").innerText = "Please select a language and ensure a summary is available.";
+                return;
+            }
+            console.log("Translate called with lang:", lang, "Text:", summary);
+            try {
+                const response = await fetch("/translate", {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ text: summary, lang: lang })
+                });
+                const data = await response.json();
+                console.log("Translate response:", JSON.stringify(data, null, 2));
+                if (data.translation) {
+                    document.getElementById("translation").innerText = data.translation;
+                } else {
+                    document.getElementById("translation").innerText = "Error: " + (data.error || "Unknown error");
+                }
+            } catch (error) {
+                console.error("Translate error:", error);
+                document.getElementById("translation").innerText = "Failed to translate: " + error.message;
+            }
+        }
+    </script>
+</body>
+</html>

requirements.txt ADDED Viewed

	@@ -0,0 +1,4 @@

+fastapi
+uvicorn
+transformers
+torch