Update app.py

app.py

@@ -19,18 +19,38 @@ import shutil
 from pathlib import Path
 import py_text_scan
 from sqlalchemy import create_engine, Column, Integer, String, Boolean, Text, DateTime
+# --- FIX: Added the missing import below ---
+from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker, Session
 from passlib.context import CryptContext
 import datetime
 
-# --- Database and other setup remains the same ---
+# --- Database Setup (SQLite) ---
 DATABASE_URL = "sqlite:///./test.db"
 engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+# This line will now work correctly
 Base = declarative_base()
 
-# ... (Database Models, Pydantic Schemas, Auth functions remain the same) ...
-# NOTE: To keep the code brief, repeating the unchanged parts.
+# --- Database Models ---
+class UserModel(Base):
+    __tablename__ = "users"
+    id = Column(Integer, primary_key=True, index=True)
+    username = Column(String, unique=True, index=True)
+    email = Column(String, unique=True, index=True)
+    hashed_password = Column(String)
+    is_active = Column(Boolean, default=True)
+    is_admin = Column(Boolean, default=False)
+
+class FeedbackModel(Base):
+    __tablename__ = "feedback"
+    id = Column(Integer, primary_key=True, index=True)
+    username = Column(String)
+    comment = Column(Text)
+    created_at = Column(DateTime, default=datetime.datetime.utcnow)
+
+Base.metadata.create_all(bind=engine)
+
 
 class OCRResponse(BaseModel):
     sakshi_output: str
@@ -110,7 +130,7 @@ def process_image(image_array, use_keras: bool, use_py_text_scan: bool):
     cv2.imwrite(word_detection_path, word_detected_img)
     session_files['word_detection'] = word_detection_path
 
-    # --- MODIFICATION: Conditional Keras Model Prediction ---
+    # --- Conditional Keras Model Prediction ---
     pred_label = "Keras model disabled by user"
     if use_keras:
         try:
@@ -126,7 +146,7 @@ def process_image(image_array, use_keras: bool, use_py_text_scan: bool):
         except Exception as e:
             pred_label = f"Keras Error: {str(e)}"
     
-    # --- MODIFICATION: Conditional py_text_scan Execution ---
+    # --- Conditional py_text_scan Execution ---
     sakshi_output = "py_text_scan disabled by user"
     if use_py_text_scan:
         with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
@@ -141,7 +161,6 @@ def process_image(image_array, use_keras: bool, use_py_text_scan: bool):
     }
 
 # --- API Endpoints ---
-# MODIFIED: Endpoint now takes query parameters to control models
 @app.post("/process/", response_model=OCRResponse)
 async def process(
     file: UploadFile = File(...),
@@ -185,7 +204,5 @@ async def process(
         if os.path.exists(temp_file_path):
             os.unlink(temp_file_path)
 
-# ... (other endpoints like /word-detection/ can remain as they are or be removed if not needed) ...
-
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)