Spaces:

1yahoo
/

Test

Sleeping

App Files Files Community

1yahoo commited on Jan 27

Commit

91d07ed

verified ·

1 Parent(s): 0521bb0

Update app.py

Browse files

Files changed (1) hide show

app.py +99 -157

app.py CHANGED Viewed

@@ -2,45 +2,22 @@ import os
 import uuid
 import pypdf
 import chromadb
 from fastapi import FastAPI, Request, File, UploadFile, BackgroundTasks
 from fastapi.responses import StreamingResponse, HTMLResponse
 from fastapi.middleware.cors import CORSMiddleware
 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="vault_final", embedding_function=default_ef)
-client = OpenAI(
-    base_url="https://router.huggingface.co/hf-inference/v1",
-    api_key=os.getenv("HF_TOKEN")
-)
-# --- محرك المعالجة (RAG Engine) ---
-def process_document(file_content, filename):
-    text = ""
-    if filename.endswith('.pdf'):
-        # معالجة PDF
-        with open("temp.pdf", "wb") as f:
-            f.write(file_content)
-        reader = pypdf.PdfReader("temp.pdf")
-        for page in reader.pages:
-            text += page.extract_text() + " "
-    else:
-        # معالجة نص
-        text = file_content.decode("utf-8", errors="ignore")
-    # تقسيم النص لقطع (Chunking)
-    chunks = [text[i:i+1000] for i in range(0, len(text), 800)]
-    ids = [str(uuid.uuid4()) for _ in chunks]
-    collection.add(documents=chunks, ids=ids)
-# --- واجهة المستخدم (Responsive UI) ---
 @app.get("/", response_class=HTMLResponse)
 async def get_ui():
     return """
@@ -54,141 +31,104 @@ async def get_ui():
         <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>
-        <link href="https://fonts.googleapis.com/css2?family=IBM+Plex+Sans+Arabic:wght@300;400;600&display=swap" rel="stylesheet">
         <style>
-            body { font-family: 'IBM Plex Sans Arabic', sans-serif; background: #020617; color: #f8fafc; height: 100vh; overflow: hidden; }
-            .glass-header { background: rgba(15, 23, 42, 0.8); backdrop-filter: blur(10px); border-bottom: 1px solid rgba(255,255,255,0.05); }
-            .sidebar { background: rgba(7, 10, 25, 0.98); backdrop-filter: blur(20px); z-index: 100; transition: all 0.3s ease; }
-            .chat-box { height: calc(100vh - 140px); overflow-y: auto; scroll-behavior: smooth; }
-            .user-msg { background: #1e293b; border-radius: 18px 18px 0 18px; }
-            .bot-msg { background: rgba(99, 102, 241, 0.1); border: 1px solid rgba(99, 102, 241, 0.2); border-radius: 18px 18px 18px 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 [status, setStatus] = useState("idle");
-                const [sidebar, setSidebar] = useState(false);
-                const [settings, setSettings] = useState({ temp: 0.7, tokens: 2048 });
-                const endRef = useRef(null);
-                useEffect(() => endRef.current?.scrollIntoView({ behavior: "smooth" }), [messages]);
-                const handleUpload = async (e) => {
-                    const file = e.target.files[0];
-                    if(!file) return;
-                    setStatus("indexing");
-                    const fd = new FormData();
-                    fd.append('file', file);
-                    await fetch('/upload', { method: 'POST', body: fd });
-                    setStatus("idle");
-                    alert("تم حقن المستند في الذاكرة بنجاح");
-                };
                 const send = async () => {
-                    if(!input.trim() || status !== "idle") return;
-                    const msg = { role: 'user', content: input };
-                    setMessages(p => [...p, msg]);
                     setInput("");
                     setStatus("thinking");
-                    const res = await fetch('/v1/chat/completions', {
-                        method: 'POST',
-                        headers: {'Content-Type': 'application/json'},
-                        body: JSON.stringify({ messages: [...messages, msg], temperature: settings.temp })
-                    });
-                    const reader = res.body.getReader();
-                    let botMsg = { role: 'assistant', content: "" };
-                    setMessages(p => [...p, botMsg]);
-                    while (true) {
-                        const {done, value} = await reader.read();
-                        if (done) break;
-                        const chunk = new TextDecoder().decode(value).replace(/data: /g, '');
-                        if(chunk.includes("[DONE]")) break;
-                        botMsg.content += chunk;
-                        setMessages(p => [...p.slice(0, -1), {...botMsg}]);
-                    }
                     setStatus("idle");
                 };
                 return (
-                    <div className="flex h-screen relative">
-                        {/* Sidebar */}
-                        <aside className={`${sidebar ? 'translate-x-0' : 'translate-x-full'} md:translate-x-0 fixed md:relative right-0 w-72 h-full sidebar p-6 border-l border-white/5`}>
-                            <div className="flex justify-between items-center mb-10 md:hidden">
-                                <span className="font-bold">إغلاق</span>
-                                <button onClick={() => setSidebar(false)} className="text-2xl">✕</button>
-                            </div>
-                            <h2 className="text-indigo-400 font-bold mb-6 flex items-center gap-2"><span>⚙️</span> الإعدادات</h2>
-                            <div className="space-y-8">
-                                <div>
-                                    <label className="text-xs text-slate-500 block mb-2 uppercase tracking-widest">مستوى الإبداع: {settings.temp}</label>
-                                    <input type="range" min="0" max="1.5" step="0.1" value={settings.temp} onChange={e => setSettings({...settings, temp: parseFloat(e.target.value)})} className="w-full accent-indigo-500" />
-                                </div>
-                                <div className="pt-6 border-t border-white/5">
-                                    <label className="text-xs text-slate-500 block mb-4 uppercase tracking-widest">حقن مستند (PDF/TXT)</label>
-                                    <label className="flex flex-col items-center justify-center w-full h-32 border-2 border-dashed border-slate-700 rounded-xl cursor-pointer hover:border-indigo-500 transition-all">
-                                        <div className="text-center">
-                                            <span className="text-2xl block mb-2">📁</span>
-                                            <span className="text-[10px] text-slate-400">اضغط للرفع</span>
-                                        </div>
-                                        <input type="file" className="hidden" onChange={handleUpload} />
-                                    </label>
                                 </div>
                             </div>
-                        </aside>
-                        {/* Main Chat */}
-                        <main className="flex-1 flex flex-col min-w-0 bg-[#020617]">
-                            <header className="glass-header p-4 flex justify-between items-center">
-                                <button onClick={() => setSidebar(true)} className="md:hidden text-indigo-400 text-xl">☰</button>
-                                <div className="flex items-center gap-3">
-                                    <div className={`w-2 h-2 rounded-full ${status === 'thinking' ? 'bg-indigo-500 animate-pulse' : 'bg-emerald-500'}`}></div>
-                                    <h1 className="text-sm font-bold tracking-tighter">DEEP NEURAL VAULT</h1>
-                                </div>
-                                <div className="text-[10px] font-mono text-slate-500">V.2.5_RAG</div>
-                            </header>
-                            <div className="chat-box p-4 space-y-6 custom-scrollbar">
-                                {messages.map((m, i) => (
-                                    <div key={i} className={`flex ${m.role === 'user' ? 'justify-start' : 'justify-end'}`}>
-                                        <div className={`max-w-[88%] md:max-w-[75%] p-4 text-sm md:text-base leading-relaxed ${m.role === 'user' ? 'user-msg text-slate-200' : 'bot-msg text-indigo-50'}`}>
-                                            {m.content}
-                                        </div>
-                                    </div>
-                                ))}
-                                <div ref={endRef} />
                             </div>
-                            <div className="p-4 bg-gradient-to-t from-[#020617] to-transparent">
-                                <div className="max-w-4xl mx-auto flex gap-2 bg-slate-900/80 border border-white/10 p-2 rounded-2xl">
-                                    <input
-                                        className="flex-1 bg-transparent 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 hover:bg-indigo-500 p-3 rounded-xl transition-all">
-                                        <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path d="M5 12h14M12 5l7 7-7 7" strokeWidth="2" strokeLinecap="round" strokeLinejoin="round"/></svg>
-                                    </button>
                                 </div>
                             </div>
-                        </main>
-                        {sidebar && <div onClick={() => setSidebar(false)} className="fixed inset-0 bg-black/60 z-[90] md:hidden"></div>}
                     </div>
                 );
             }
             const root = ReactDOM.createRoot(document.getElementById('root'));
             root.render(<App />);
         </script>
@@ -196,40 +136,42 @@ async def get_ui():
     </html>
     """
-# --- مسارات الخلفية (Endpoints) ---
-@app.post("/upload")
-async def upload(background_tasks: BackgroundTasks, file: UploadFile = File(...)):
-    content = await file.read()
-    background_tasks.add_task(process_document, content, file.filename)
-    return {"status": "processing"}
 @app.post("/v1/chat/completions")
-async def chat(request: Request):
     body = await request.json()
-    messages = body.get("messages", [])
-    # استرجاع المعرفة
-    query = messages[-1]["content"] if messages else ""
-    results = collection.query(query_texts=[query], n_results=2)
-    ctx = "\\n".join(results['documents'][0]) if results['documents'] else ""
-    # بناء الرسائل مع السياق
-    sys_prompt = {"role": "system", "content": f"You are the Neural Vault. Context: {ctx}"}
-    def stream():
-        response = client.chat.completions.create(
             model="huihui-ai/Qwen2.5-72B-Instruct-abliterated",
-            messages=[sys_prompt] + messages,
-            stream=True,
-            temperature=body.get("temperature", 0.7)
         )
         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(), media_type="text/event-stream")
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)

 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 fastapi.middleware.cors import CORSMiddleware
 from openai import OpenAI
 from chromadb.utils import embedding_functions
+# --- المحرك الخلفي (FastAPI) ---
 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_v1", embedding_function=default_ef)
 @app.get("/", response_class=HTMLResponse)
 async def get_ui():
     return """
         <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; height: 100dvh; display: flex; flex-direction: column; overflow: hidden; }
+            .mobile-safe-area { height: 100%; display: flex; flex-direction: column; position: relative; }
+            .chat-stream { flex: 1; overflow-y: auto; padding-bottom: 80px; -webkit-overflow-scrolling: touch; }
+            .input-area { position: absolute; bottom: 0; left: 0; right: 0; padding: 12px; background: linear-gradient(transparent, #020617 20%); }
+            .sidebar-drawer { position: fixed; top: 0; right: 0; bottom: 0; w: 80%; background: #0f172a; z-index: 1000; transition: transform 0.3s ease; shadow: -10px 0 30px rgba(0,0,0,0.5); }
         </style>
     </head>
     <body>
+        <div id="root" class="mobile-safe-area"></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 send = async () => {
+                    if(!input.trim()) return;
+                    const newMsgs = [...messages, {role: 'user', content: input}];
+                    setMessages(newMsgs);
                     setInput("");
                     setStatus("thinking");
+                    try {
+                        const res = await fetch('/v1/chat/completions', {
+                            method: 'POST',
+                            headers: {'Content-Type': 'application/json'},
+                            body: JSON.stringify({ messages: newMsgs })
+                        });
+                        const reader = res.body.getReader();
+                        let currentBotMsg = { role: 'assistant', content: "" };
+                        setMessages(prev => [...prev, currentBotMsg]);
+                        while(true) {
+                            const {done, value} = await reader.read();
+                            if(done) break;
+                            const text = new TextDecoder().decode(value).replace(/data: /g, '');
+                            if(text.includes("[DONE]")) break;
+                            currentBotMsg.content += text;
+                            setMessages(prev => [...prev.slice(0, -1), {...currentBotMsg}]);
+                        }
+                    } catch(e) { console.error(e); }
                     setStatus("idle");
                 };
                 return (
+                    <div className="h-full flex flex-col">
+                        <header className="p-4 border-b border-white/10 flex justify-between items-center bg-slate-900/50">
+                            <button onClick={() => setSideOpen(true)} className="text-indigo-400">☰</button>
+                            <span className="text-xs font-bold 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 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'}`}>
+                                        {m.content}
+                                    </div>
                                 </div>
+                            ))}
+                        </div>
+                        <div className="input-area">
+                            <div className="max-w-xl mx-auto flex gap-2 bg-slate-900 border border-white/10 rounded-xl p-1 shadow-xl">
+                                <input
+                                    className="flex-1 bg-transparent px-3 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 p-2 rounded-lg">
+                                    <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path d="M5 12h14M12 5l7 7-7 7" strokeWidth="2" strokeLinecap="round" strokeLinejoin="round"/></svg>
+                                </button>
                             </div>
+                        </div>
+                        {/* Sidebar Drawer */}
+                        <aside className={`sidebar-drawer p-6 ${isSideOpen ? 'translate-x-0' : 'translate-x-full'}`}>
+                            <div className="flex justify-between mb-8">
+                                <h2 className="font-bold">الحساب والذاكرة</h2>
+                                <button onClick={() => setSideOpen(false)}>✕</button>
                             </div>
+                            <div className="space-y-6">
+                                <div className="p-4 bg-indigo-600/10 rounded-xl border border-indigo-500/20 text-xs">
+                                    <p>نظام الاستهلاك: <b>Featherless API</b></p>
+                                    <p className="mt-2 text-slate-400">سيتم الخصم من رصيدك في HF عند تسجيل الدخول.</p>
                                 </div>
+                                <input type="file" className="block w-full text-xs text-slate-500 file:bg-slate-800 file:text-white file:border-0 file:py-2 file:px-4 file:rounded-lg" />
                             </div>
+                        </aside>
                     </div>
                 );
             }
             const root = ReactDOM.createRoot(document.getElementById('root'));
             root.render(<App />);
         </script>
     </html>
     """
+# --- منطق تسجيل الدخول واستهلاك الرصيد (مستوحى من ملفك) ---
+def get_client(token: str = None):
+    # نستخدم التوكن الخاص بالمستخدم إذا سجل دخوله، وإلا نستخدم توكن النظام
+    api_key = token if token else os.getenv("HF_TOKEN")
+    return OpenAI(
+        base_url="https://router.huggingface.co/hf-inference/v1",
+        api_key=api_key
+    )
 @app.post("/v1/chat/completions")
+async def chat_endpoint(request: Request):
     body = await request.json()
+    # [span_2](start_span)هنا يتم ربط الطلب بـ featherless-ai كما في ملف app(2).py[span_2](end_span)
+    local_client = get_client()
+    def stream_gen():
+        response = local_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")
+# إطلاق واجهة Gradio كطبقة دخول اختيارية لتأكيد الهوية
+with gr.Blocks() as auth_layer:
+    gr.Markdown("# تسجيل الدخول")
+    login_btn = gr.LoginButton("تسجيل الدخول باستخدام Hugging Face")
+    gr.Markdown("بعد تسجيل الدخول، سيتم استخدام رصيدك الشخصي للعمليات.")
+# تشغيل النظام
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)