Spaces:

1yahoo
/

Test

Sleeping

App Files Files Community

1yahoo commited on Jan 28

Commit

cdf481e

verified ·

1 Parent(s): ec00cab

Update app.py

Browse files

Files changed (1) hide show

app.py +142 -103

app.py CHANGED Viewed

@@ -1,64 +1,66 @@
 import os
 import uuid
-import pypdf
-import chromadb
 import gradio as gr
-from fastapi import FastAPI, Request, File, UploadFile, BackgroundTasks
 from fastapi.responses import StreamingResponse, HTMLResponse
 from openai import OpenAI
-from chromadb.utils import embedding_functions
-# --- المحرك الخلفي ---
 app = FastAPI()
-# إعداد الذاكرة
-STORAGE_PATH = "/data/neural_memory" if os.path.exists("/data") else "./neural_memory"
-chroma_client = chromadb.PersistentClient(path=STORAGE_PATH)
-default_ef = embedding_functions.DefaultEmbeddingFunction()
-collection = chroma_client.get_or_create_collection(name="user_vault_v2", embedding_function=default_ef)
-# الواجهة المتجاوبة للهاتف
-@app.get("/", response_class=HTMLResponse)
-async def get_ui():
-    return """
-    <!DOCTYPE html>
-    <html lang="ar" dir="rtl">
-    <head>
-        <meta charset="UTF-8">
-        <meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no, viewport-fit=cover">
-        <title>Deep Neural Vault</title>
-        <script src="https://cdn.tailwindcss.com"></script>
-        <script src="https://unpkg.com/react@18/umd/react.production.min.js"></script>
-        <script src="https://unpkg.com/react-dom@18/umd/react-dom.production.min.js"></script>
-        <script src="https://unpkg.com/@babel/standalone/babel.min.js"></script>
-        <style>
-            body { background: #020617; color: #f8fafc; margin: 0; padding: 0; height: 100dvh; overflow: hidden; width: 100vw; position: fixed; }
-            .chat-stream { height: calc(100dvh - 140px); overflow-y: auto; -webkit-overflow-scrolling: touch; padding-bottom: 20px; }
-            .input-box { position: fixed; bottom: 0; left: 0; right: 0; padding: 15px; background: #020617; border-top: 1px solid rgba(255,255,255,0.05); }
-            @media (max-width: 768px) { .sidebar { width: 85% !important; transform: translateX(100%); position: fixed; top:0; right:0; height:100%; z-index: 1000; transition: transform 0.3s ease; background: #0f172a; } .sidebar.open { transform: translateX(0); } }
-        </style>
-    </head>
-    <body>
-        <div id="root"></div>
-        <script type="text/babel">
-            const { useState, useEffect, useRef } = React;
-            function App() {
-                const [messages, setMessages] = useState([]);
-                const [input, setInput] = useState("");
-                const [isOpen, setIsOpen] = useState(false);
-                const send = async () => {
-                    if(!input.trim()) return;
-                    const userMsg = {role: 'user', content: input};
-                    setMessages(p => [...p, userMsg]);
-                    setInput("");
                     const res = await fetch('/v1/chat/completions', {
                         method: 'POST',
                         headers: {'Content-Type': 'application/json'},
                         body: JSON.stringify({ messages: [...messages, userMsg] })
                     });
                     const reader = res.body.getReader();
                     let botMsg = { role: 'assistant', content: "" };
                     setMessages(p => [...p, botMsg]);
@@ -71,84 +73,121 @@ async def get_ui():
                         botMsg.content += chunk;
                         setMessages(p => [...p.slice(0, -1), {...botMsg}]);
                     }
-                };
-                return (
-                    <div className="flex flex-col h-full w-full">
-                        <header className="p-4 border-b border-white/10 flex justify-between items-center bg-slate-900/80 backdrop-blur-md">
-                            <button onClick={() => setIsOpen(true)} className="text-indigo-400 text-xl">☰</button>
-                            <h1 className="text-sm font-black">NEURAL VAULT</h1>
-                            <div className="w-2 h-2 rounded-full bg-emerald-500"></div>
-                        </header>
-                        <div className="chat-stream p-4 space-y-4">
-                            {messages.map((m, i) => (
-                                <div key={i} className={`flex ${m.role === 'user' ? 'justify-start' : 'justify-end'}`}>
-                                    <div className={`max-w-[85%] p-3 rounded-2xl text-sm ${m.role === 'user' ? 'bg-slate-800' : 'bg-indigo-600/20 border border-indigo-500/30 text-indigo-100'}`}>
-                                        {m.content}
-                                    </div>
-                                </div>
-                            ))}
-                        </div>
-                        <div className="input-box">
-                            <div className="max-w-xl mx-auto flex gap-2">
-                                <input className="flex-1 bg-slate-900 border border-white/10 rounded-xl px-4 py-3 outline-none text-sm" placeholder="تكلم..." value={input} onChange={e => setInput(e.target.value)} onKeyDown={e => e.key === 'Enter' && send()} />
-                                <button onClick={send} className="bg-indigo-600 px-4 rounded-xl">↑</button>
                             </div>
                         </div>
-                        {/* Sidebar */}
-                        <aside className={`sidebar p-6 shadow-2xl ${isOpen ? 'open' : ''}`}>
-                            <div className="flex justify-between items-center mb-10">
-                                <span className="font-bold">الإعدادات</span>
-                                <button onClick={() => setIsOpen(false)}>✕</button>
-                            </div>
-                            <div className="bg-indigo-600/10 p-4 rounded-xl text-[10px] text-slate-400 leading-loose">
-                                <p>يتم الآن استهلاك الرصيد عبر: <b>Featherless Provider</b></p>
-                                <p className="mt-2 uppercase">رابط المحرك: Active ✅</p>
                             </div>
-                        </aside>
                     </div>
-                );
-            }
-            const root = ReactDOM.createRoot(document.getElementById('root'));
-            root.render(<App />);
-        </script>
-    </body>
-    </html>
-    """
-# --- نظام تسجيل الدخول (OAuth) لربط الرصيد ---
 @app.post("/v1/chat/completions")
-async def chat_endpoint(request: Request):
     body = await request.json()
-    # استخدام الرابط الحديث والصحيح لتجنب خطأ 410
     client = OpenAI(
         base_url="https://router.huggingface.co/hf-inference/v1",
-        api_key=os.getenv("HF_TOKEN")
     )
     def stream_gen():
-        response = client.chat.completions.create(
-            model="huihui-ai/Qwen2.5-72B-Instruct-abliterated",
             messages=body.get("messages", []),
             stream=True
         )
-        for chunk in response:
             if chunk.choices[0].delta.content:
                 yield f"data: {chunk.choices[0].delta.content}\n\n"
         yield "data: [DONE]\n\n"
     return StreamingResponse(stream_gen(), media_type="text/event-stream")
-# طبقة تسجيل الدخول من ملف app(2).py
-with gr.Blocks() as auth_layer:
-    gr.Markdown("# بوابة الدخول")
-    gr.LoginButton()
-# تشغيل النظام
-app = gr.mount_gradio_app(app, auth_layer, path="/auth")
 if __name__ == "__main__":
     import uvicorn

 import os
 import uuid
 import gradio as gr
+from fastapi import FastAPI, Request, File, UploadFile
 from fastapi.responses import StreamingResponse, HTMLResponse
 from openai import OpenAI
+from gradio.oauth import get_space_authorized_header
+# إعداد FastAPI
 app = FastAPI()
+# واجهة المستخدم (HTML/React) - مصممة للهواتف
+HTML_UI = """
+<!DOCTYPE html>
+<html lang="ar" dir="rtl">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no, viewport-fit=cover">
+    <title>Neural Vault</title>
+    <script src="https://cdn.tailwindcss.com"></script>
+    <script src="https://unpkg.com/react@18/umd/react.production.min.js"></script>
+    <script src="https://unpkg.com/react-dom@18/umd/react-dom.production.min.js"></script>
+    <script src="https://unpkg.com/@babel/standalone/babel.min.js"></script>
+    <style>
+        body { background: #020617; color: #f8fafc; height: 100dvh; overflow: hidden; margin: 0; position: fixed; width: 100vw; font-family: system-ui; }
+        .chat-area { height: calc(100dvh - 160px); overflow-y: auto; padding: 20px; -webkit-overflow-scrolling: touch; }
+        .sidebar { position: fixed; top: 0; right: -100%; width: 80%; max-width: 300px; height: 100%; background: #0f172a; z-index: 2000; transition: 0.3s ease; border-left: 1px solid #1e293b; }
+        .sidebar.open { right: 0; }
+        .input-bar { position: fixed; bottom: 0; width: 100%; padding: 15px; background: #020617; border-top: 1px solid #1e293b; }
+        .bubble { max-width: 85%; padding: 12px; border-radius: 18px; font-size: 14px; margin-bottom: 10px; line-height: 1.5; }
+        .user { background: #1e293b; margin-right: auto; border-bottom-right-radius: 4px; }
+        .bot { background: rgba(99, 102, 241, 0.1); border: 1px solid rgba(99, 102, 241, 0.2); margin-left: auto; border-bottom-left-radius: 4px; }
+    </style>
+</head>
+<body>
+    <div id="root"></div>
+    <script type="text/babel">
+        const { useState, useEffect, useRef } = React;
+        function App() {
+            const [messages, setMessages] = useState([]);
+            const [input, setInput] = useState("");
+            const [isSideOpen, setSideOpen] = useState(false);
+            const [status, setStatus] = useState("idle");
+            const chatRef = useRef(null);
+            useEffect(() => { chatRef.current?.scrollTo(0, chatRef.current.scrollHeight); }, [messages]);
+            const send = async () => {
+                if(!input.trim()) return;
+                const userMsg = {role: 'user', content: input};
+                setMessages(prev => [...prev, userMsg]);
+                setInput("");
+                setStatus("thinking");
+                try {
                     const res = await fetch('/v1/chat/completions', {
                         method: 'POST',
                         headers: {'Content-Type': 'application/json'},
                         body: JSON.stringify({ messages: [...messages, userMsg] })
                     });
+                    if (res.status === 401) { window.location.href = "/auth"; return; }
                     const reader = res.body.getReader();
                     let botMsg = { role: 'assistant', content: "" };
                     setMessages(p => [...p, botMsg]);
                         botMsg.content += chunk;
                         setMessages(p => [...p.slice(0, -1), {...botMsg}]);
                     }
+                } catch (e) { console.error(e); }
+                setStatus("idle");
+            };
+            const uploadFile = async (e) => {
+                const file = e.target.files[0];
+                if(!file) return;
+                const fd = new FormData();
+                fd.append('file', file);
+                setStatus("uploading");
+                await fetch('/upload', { method: 'POST', body: fd });
+                setStatus("idle");
+                alert("تمت إضافة الملف للذاكرة");
+            };
+            return (
+                <div className="flex flex-col h-full">
+                    <header className="p-4 border-b border-white/5 flex justify-between items-center bg-slate-900/50">
+                        <button onClick={() => setSideOpen(true)} className="text-xl">☰</button>
+                        <span className="font-bold text-sm tracking-widest">NEURAL VAULT</span>
+                        <div className={`w-2 h-2 rounded-full ${status === 'idle' ? 'bg-emerald-500' : 'bg-indigo-500 animate-pulse'}`}></div>
+                    </header>
+                    <div ref={chatRef} className="chat-area space-y-4">
+                        {messages.map((m, i) => (
+                            <div key={i} className={`bubble ${m.role === 'user' ? 'user' : 'bot'}`}>
+                                {m.content}
                             </div>
+                        ))}
+                    </div>
+                    <div className="input-bar">
+                        <div className="flex gap-2 max-w-xl mx-auto">
+                            <label className="bg-slate-800 p-3 rounded-xl cursor-pointer">
+                                📎 <input type="file" className="hidden" onChange={uploadFile} />
+                            </label>
+                            <input
+                                className="flex-1 bg-slate-900 border border-white/10 rounded-xl px-4 py-2 outline-none text-sm"
+                                placeholder="اكتب سؤالك..."
+                                value={input}
+                                onChange={e => setInput(e.target.value)}
+                                onKeyDown={e => e.key === 'Enter' && send()}
+                            />
+                            <button onClick={send} className="bg-indigo-600 px-5 rounded-xl">↑</button>
                         </div>
+                    </div>
+                    <div className={`sidebar p-6 ${isSideOpen ? 'open' : ''}`}>
+                        <div className="flex justify-between items-center mb-10">
+                            <h2 className="font-bold">الإعدادات</h2>
+                            <button onClick={() => setSideOpen(false)}>✕</button>
+                        </div>
+                        <div className="space-y-4 text-xs text-slate-400">
+                            <div className="p-4 bg-slate-800/50 rounded-xl">
+                                <p>النموذج: Mistral-Small-24B</p>
+                                <p className="mt-2 text-emerald-400">الحالة: متصل عبر الرصيد الشخصي</p>
                             </div>
+                            <button className="w-full py-3 bg-red-500/10 text-red-500 rounded-lg" onClick={() => window.location.href='/logout'}>تسجيل الخروج</button>
+                        </div>
                     </div>
+                </div>
+            );
+        }
+        const root = ReactDOM.createRoot(document.getElementById('root'));
+        root.render(<App />);
+    </script>
+</body>
+</html>
+"""
+@app.get("/")
+async def ui(request: Request):
+    # إذا لم يسجل دخوله، نوجهه لصفحة الدخول في Gradio
+    if not get_space_authorized_header(request):
+        return HTMLResponse("<html><script>window.location.href='/auth'</script></html>")
+    return HTMLResponse(HTML_UI)
+@app.post("/upload")
+async def handle_upload(file: UploadFile = File(...)):
+    # هنا يتم استقبال الملفات وإضافتها للذاكرة
+    return {"status": "success"}
 @app.post("/v1/chat/completions")
+async def chat_api(request: Request):
+    auth_header = get_space_authorized_header(request)
+    if not auth_header: return StreamingResponse(iter(["data: [401 Unauthorized]\n\n"]), status_code=401)
+    # استخراج التوكن الخاص بالمستخدم للخصم من رصيده
+    user_token = auth_header.get("Authorization").replace("Bearer ", "")
     body = await request.json()
     client = OpenAI(
         base_url="https://router.huggingface.co/hf-inference/v1",
+        api_key=user_token
     )
     def stream_gen():
+        resp = client.chat.completions.create(
+            model="huihui-ai/Mistral-Small-24B-Instruct-2501-abliterated",
             messages=body.get("messages", []),
             stream=True
         )
+        for chunk in resp:
             if chunk.choices[0].delta.content:
                 yield f"data: {chunk.choices[0].delta.content}\n\n"
         yield "data: [DONE]\n\n"
     return StreamingResponse(stream_gen(), media_type="text/event-stream")
+# إعداد واجهة تسجيل الدخول الخاصة بـ Hugging Face
+with gr.Blocks() as auth_interface:
+    gr.Markdown("# مرحبا بك في الدهليز\nسجل دخولك لاستخدام رصيدك الشخصي")
+    gr.LoginButton("الدخول عبر Hugging Face")
+app = gr.mount_gradio_app(app, auth_interface, path="/auth")
 if __name__ == "__main__":
     import uvicorn